Skip to content

Commit 27a4a3d

Browse files
committed
Move kwarg
1 parent ea4a372 commit 27a4a3d

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/link/pytorch/test_slinalg.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,5 @@ def test_batched_mvnormal_logp_and_dlogp(cov_batch_shape):
2121

2222
chol_cov = cholesky(cov, lower=True, on_error="raise")
2323

24-
fn = pytensor.function([cov], [chol_cov])
25-
assert np.all(
26-
np.isclose(fn(test_values, mode="PYTORCH"), np.linalg.cholesky(test_values))
27-
)
24+
fn = pytensor.function([cov], [chol_cov], mode="PYTORCH")
25+
assert np.all(np.isclose(fn(test_values), np.linalg.cholesky(test_values)))

0 commit comments

Comments
 (0)