Fix autocontrast test

hertschuh · Sep 18, 2024 · dd80697 · dd80697
1 parent dd5f870
commit dd80697
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/auto_contrast_test.py b/keras/src/layers/preprocessing/image_preprocessing/auto_contrast_test.py
@@ -90,6 +90,6 @@ def test_auto_contrast_properly_converts_value_range(self):
 
         layer = layers.AutoContrast(value_range=(0, 1))
         ys = layer(img)
-
-        self.assertTrue(np.any(ops.convert_to_numpy(ys[0]) == 0.0))
-        self.assertTrue(np.any(ops.convert_to_numpy(ys[0]) == 1.0))
+        self.assertAllClose(
+            ops.convert_to_numpy(ys[0]), np.array([[[0.0]], [[1]]])
+        )