We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ee25aae commit 484f725Copy full SHA for 484f725
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