Skip to content

Commit 2d061a5

Browse files
committed
Modified SVD test
1 parent 326f239 commit 2d061a5

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/link/pytorch/test_nlinalg.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,7 @@ def test_svd(compute_uv, full_matrices, matrix_test):
6060
out = pt_nla.svd(x, full_matrices=full_matrices, compute_uv=compute_uv)
6161
out_fg = FunctionGraph([x], out if isinstance(out, list) else [out])
6262

63-
def assert_fn(x, y):
64-
np.testing.assert_allclose(x, y, rtol=1e-3)
65-
66-
compare_pytorch_and_py(out_fg, [test_value], assert_fn=assert_fn)
63+
compare_pytorch_and_py(out_fg, [test_value])
6764

6865

6966
def test_pinv():

0 commit comments

Comments
 (0)