Fix torch GPU CI, I suppose...

keras-team · Nov 23, 2024 · a503a16 · a503a16
1 parent 28d39c0
commit a503a16
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_translation_test.py b/keras/src/layers/preprocessing/image_preprocessing/random_translation_test.py
@@ -426,9 +426,7 @@ def test_random_flip_tf_data_bounding_boxes(
         )
 
         transformation = {
-            "translations": backend_utils.convert_tf_tensor(
-                np.array(translation)
-            ),
+            "translations": np.array(translation),
             "input_shape": image_shape,
         }
 

diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_zoom_test.py b/keras/src/layers/preprocessing/image_preprocessing/random_zoom_test.py
@@ -251,8 +251,8 @@ def test_random_flip_tf_data_bounding_boxes(self, zoom, expected_boxes):
         )
 
         transformation = {
-            "height_zoom": backend_utils.convert_tf_tensor(np.array(zoom[0])),
-            "width_zoom": backend_utils.convert_tf_tensor(np.array(zoom[1])),
+            "height_zoom": np.array(zoom[0]),
+            "width_zoom": np.array(zoom[1]),
             "input_shape": image_shape,
         }