Skip to content

Commit 35444df

Browse files
Respect check_finite flag in cholesky
1 parent 6f1e115 commit 35444df

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pytensor/tensor/slinalg.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,9 @@ def cholesky(
201201
202202
"""
203203

204-
return Blockwise(Cholesky(lower=lower, on_error=on_error))(x)
204+
return Blockwise(
205+
Cholesky(lower=lower, on_error=on_error, check_finite=check_finite)
206+
)(x)
205207

206208

207209
class SolveBase(Op):

0 commit comments

Comments
 (0)