Modified SVD test

pymc-devs · Jul 23, 2024 · 2d061a5 · 2d061a5
1 parent 326f239
commit 2d061a5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/link/pytorch/test_nlinalg.py b/tests/link/pytorch/test_nlinalg.py
@@ -60,10 +60,7 @@ def test_svd(compute_uv, full_matrices, matrix_test):
     out = pt_nla.svd(x, full_matrices=full_matrices, compute_uv=compute_uv)
     out_fg = FunctionGraph([x], out if isinstance(out, list) else [out])
 
-    def assert_fn(x, y):
-        np.testing.assert_allclose(x, y, rtol=1e-3)
-
-    compare_pytorch_and_py(out_fg, [test_value], assert_fn=assert_fn)
+    compare_pytorch_and_py(out_fg, [test_value])
 
 
 def test_pinv():