Merge pull request #213 from ev-br/torch_allclose

MAINT: avoid a DeprecationWarning from torch
data-apis · Dec 12, 2024 · edde2e0 · edde2e0
2 parents d9df003 + 484f725
commit edde2e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vendor_test/uses_torch.py b/vendor_test/uses_torch.py
@@ -23,7 +23,7 @@ def _test_torch():
     assert isinstance(b, torch.Tensor)
     assert isinstance(res, torch.Tensor)
 
-    torch.testing.assert_allclose(res, [[1., 2., 3.]])
+    torch.testing.assert_close(res, torch.as_tensor([[1., 2., 3.]]))
 
     assert is_torch_array(res)
     assert is_torch_namespace(torch) and is_torch_namespace(torch_compat)