use assert_close

justincdavis · justincdavis · commit 5ce83b11bfef · 2025-12-04T11:10:00.000-08:00
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -4089,7 +4089,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:
