change resizing test

keras-team · Nov 19, 2023 · 0d8a432 · 0d8a432
1 parent 7ab306a
commit 0d8a432
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_cv/layers/preprocessing/resizing_test.py b/keras_cv/layers/preprocessing/resizing_test.py
@@ -25,7 +25,7 @@ def _run_output_shape_test(self, kwargs, height, width):
         kwargs.update({"height": height, "width": width})
         layer = cv_layers.Resizing(**kwargs)
 
-        inputs = tf.random.uniform((2, 5, 8, 3))
+        inputs = np.random.uniform((2, 5, 8, 3))
         outputs = layer(inputs)
         self.assertEqual(outputs.shape, (2, height, width, 3))