We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents d9df003 + 484f725 commit edde2e0Copy full SHA for edde2e0
vendor_test/uses_torch.py
@@ -23,7 +23,7 @@ def _test_torch():
23
assert isinstance(b, torch.Tensor)
24
assert isinstance(res, torch.Tensor)
25
26
- torch.testing.assert_allclose(res, [[1., 2., 3.]])
+ torch.testing.assert_close(res, torch.as_tensor([[1., 2., 3.]]))
27
28
assert is_torch_array(res)
29
assert is_torch_namespace(torch) and is_torch_namespace(torch_compat)
0 commit comments