This isn't counterproductive dev speed.

huggingface · Jul 29, 2024 · d799e80 · d799e80
1 parent 684cebd
commit d799e80
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bindings/python/tests/test_simple.py b/bindings/python/tests/test_simple.py
@@ -248,7 +248,9 @@ def test_torch_slice(self):
 
             tensor = slice_[:2]
             self.assertEqual(list(tensor.shape), [2, 5])
-            torch.testing.assert_close(tensor, A[:2], f"{tensor} != {A[:2]}")
+            if not torch.allclose(tensor, A[:2]):
+                print(f"{tensor} != {A[:2]}")
+            torch.testing.assert_close(tensor, A[:2])
 
             tensor = slice_[:, :2]
             self.assertEqual(list(tensor.shape), [10, 2])