use assert_close

justincdavis · justincdavis · commit 0019e9f220b5 · 2025-12-01T18:35:16.000-08:00
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -4036,7 +4036,7 @@ def test_functional_image_correctness(self, dimensions, kernel_size, sigma, dtyp
             actual = F.cvcuda_to_tensor(actual)
             actual = actual.squeeze(0).to(device=device)
 
-        torch.testing.assert_close(actual, expected, rtol=0, atol=1)
+        assert_close(actual, expected, rtol=0, atol=1)
 
 
 class TestGaussianNoise:
