update

tests/tensorflow/test_models_preprocessor_tf.py

@@ -40,5 +40,5 @@ def test_preprocessor(batch_size, output_size, input_tensor, expected_batches, e
     assert all(isinstance(b, tf.Tensor) for b in out)
     assert all(b.dtype == tf.float32 for b in out)
     assert all(b.shape[1:3] == output_size for b in out)
-    assert all(tf.math.reduce_all(b == expected_value) for b in out)
+    assert all(tf.math.reduce_all(tf.math.abs(b - expected_value) < 1e-6) for b in out)
     assert len(repr(processor).split("\n")) == 4

tests/tensorflow/test_transforms_tf.py

@@ -14,7 +14,7 @@ def test_resize():
     input_t = tf.cast(tf.fill([64, 64, 3], 1), dtype=tf.float32)
     out = transfo(input_t)
 
-    assert tf.reduce_all(out == 1)
+    assert tf.math.reduce_all(tf.math.abs(out - 1) < 1e-6)
     assert out.shape[:2] == output_size
     assert repr(transfo) == f"Resize(output_size={output_size}, method='bilinear')"
 
@@ -24,7 +24,7 @@ def test_resize():
 
     assert not tf.reduce_all(out == 1)
     # Asymetric padding
-    assert tf.reduce_all(out[-1] == 0) and tf.reduce_all(out[0] == 1)
+    assert tf.reduce_all(out[-1] == 0) and tf.math.reduce_all(tf.math.abs(out[0] - 1) < 1e-6)
     assert out.shape[:2] == output_size
 
     # Symetric padding