Fix GPU Tests (#2228)

keras-team · Dec 11, 2023 · 467f19d · 467f19d
1 parent e4fbbe0
commit 467f19d
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 4 deletions.
diff --git a/cloudbuild/Dockerfile → .cloudbuild/Dockerfile b/cloudbuild/Dockerfile → .cloudbuild/Dockerfile
diff --git a/cloudbuild/README.md → .cloudbuild/README.md b/cloudbuild/README.md → .cloudbuild/README.md
diff --git a/cloudbuild/cloudbuild.yaml → .cloudbuild/cloudbuild.yaml b/cloudbuild/cloudbuild.yaml → .cloudbuild/cloudbuild.yaml
diff --git a/cloudbuild/unit_test_jobs.jsonnet → .cloudbuild/unit_test_jobs.jsonnet b/cloudbuild/unit_test_jobs.jsonnet → .cloudbuild/unit_test_jobs.jsonnet
diff --git a/.kokoro/github/ubuntu/gpu/build.sh b/.kokoro/github/ubuntu/gpu/build.sh
@@ -63,8 +63,7 @@ then
       keras_cv/models/object_detection/yolo_v8 \
       keras_cv/models/object_detection_3d \
       keras_cv/models/segmentation \
-      keras_cv/models/stable_diffusion \
-      --cov=keras-cv
+      keras_cv/models/stable_diffusion
 else
    pytest --check_gpu --run_large --durations 0 \
       keras_cv/bounding_box \
@@ -78,6 +77,5 @@ else
       keras_cv/models/object_detection/yolo_v8 \
       keras_cv/models/object_detection_3d \
       keras_cv/models/segmentation \
-      keras_cv/models/stable_diffusion \
-      --cov=keras-cv
+      keras_cv/models/stable_diffusion
 fi