Skip to content

Commit b6d2abf

Browse files
Expose pt.linalg.cho_solve to users
1 parent e5278ec commit b6d2abf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytensor/tensor/slinalg.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -368,7 +368,7 @@ def cho_solve(c_and_lower, b, *, check_finite=True, b_ndim: int | None = None):
368368
Whether to check that the input matrices contain only finite numbers.
369369
Disabling may give a performance gain, but may result in problems
370370
(crashes, non-termination) if the inputs do contain infinities or NaNs.
371-
b_ndim : int
371+
b_ndim : int
372372
Whether the core case of b is a vector (1) or matrix (2).
373373
This will influence how batched dimensions are interpreted.
374374
"""
@@ -1181,4 +1181,5 @@ def block_diag(*matrices: TensorVariable):
11811181
"solve_discrete_are",
11821182
"solve_triangular",
11831183
"block_diag",
1184+
"cho_solve",
11841185
]

0 commit comments

Comments
 (0)