Skip to content

Commit edde2e0

Browse files
authored
Merge pull request #213 from ev-br/torch_allclose
MAINT: avoid a DeprecationWarning from torch
2 parents d9df003 + 484f725 commit edde2e0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vendor_test/uses_torch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def _test_torch():
2323
assert isinstance(b, torch.Tensor)
2424
assert isinstance(res, torch.Tensor)
2525

26-
torch.testing.assert_allclose(res, [[1., 2., 3.]])
26+
torch.testing.assert_close(res, torch.as_tensor([[1., 2., 3.]]))
2727

2828
assert is_torch_array(res)
2929
assert is_torch_namespace(torch) and is_torch_namespace(torch_compat)

0 commit comments

Comments
 (0)