Fix torch GPU CI

fchollet · fchollet · commit 2c95bfe4578f · 2025-01-02T09:21:13.000-08:00
diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_shear_test.py b/keras/src/layers/preprocessing/image_preprocessing/random_shear_test.py
@@ -183,9 +183,7 @@ def test_random_shear_tf_data_bounding_boxes(
         )
 
         transformation = {
-            "shear_factor": backend_utils.convert_tf_tensor(
-                np.array(translation)
-            ),
+            "shear_factor": np.array(translation),
             "input_shape": image_shape,
         }
 

Original file line number	Diff line number	Diff line change
`@@ -183,9 +183,7 @@ def test_random_shear_tf_data_bounding_boxes(`
`183`	`183`	`)`
`184`	`184`
`185`	`185`	`transformation = {`
`186`		`- "shear_factor": backend_utils.convert_tf_tensor(`
`187`		`- np.array(translation)`
`188`		`- ),`
	`186`	`+ "shear_factor": np.array(translation),`
`189`	`187`	`"input_shape": image_shape,`
`190`	`188`	`}`
`191`	`189`