Skip to content

Commit 326f239

Browse files
committed
Modified tests of nlinalg in pytorch implementation
1 parent e4c3dbd commit 326f239

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/link/pytorch/test_nlinalg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def matrix_test():
2222

2323
@pytest.mark.parametrize(
2424
"func",
25-
(pt_nla.eig, pt_nla.eigh, pt_nla.slogdet, pt_nla.MatrixInverse(), pt_nla.Det()),
25+
(pt_nla.eig, pt_nla.eigh, pt_nla.slogdet, pt_nla.inv, pt_nla.det),
2626
)
2727
def test_lin_alg_no_params(func, matrix_test):
2828
x, test_value = matrix_test
@@ -57,7 +57,7 @@ def test_qr(mode, matrix_test):
5757
def test_svd(compute_uv, full_matrices, matrix_test):
5858
x, test_value = matrix_test
5959

60-
out = pt_nla.SVD(full_matrices=full_matrices, compute_uv=compute_uv)(x)
60+
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

6363
def assert_fn(x, y):

0 commit comments

Comments
 (0)