diff --git a/test/test_functional_tensor.py b/test/test_functional_tensor.py index 42d44dfdbd9..73fa5583592 100644 --- a/test/test_functional_tensor.py +++ b/test/test_functional_tensor.py @@ -166,7 +166,7 @@ def test_rgb2hsv(self): self.assertLess(max_diff, 1e-5) s_hsv_img = scripted_fn(rgb_img) - self.assertTrue(hsv_img.allclose(s_hsv_img)) + self.assertTrue(hsv_img.allclose(s_hsv_img, atol=1e-7)) batch_tensors = self._create_data_batch(120, 100, num_samples=4, device=self.device).float() self._test_fn_on_batch(batch_tensors, F_t._rgb2hsv)