Skip to content

Commit d8fb574

Browse files
restore method dispatching
1 parent 44b36a6 commit d8fb574

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

pytensor/tensor/slinalg.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -911,7 +911,14 @@ def solve_discrete_lyapunov(
911911
A = as_tensor_variable(A)
912912
Q = as_tensor_variable(Q)
913913

914-
return cast(TensorVariable, _solve_bilinear_direct_lyapunov(A, Q))
914+
if method == "direct":
915+
return _direct_solve_discrete_lyapunov(A, Q)
916+
917+
elif method == "bilinear":
918+
return cast(TensorVariable, _solve_bilinear_direct_lyapunov(A, Q))
919+
920+
else:
921+
raise ValueError(f"Unknown method {method}")
915922

916923

917924
def solve_continuous_lyapunov(A: TensorVariable, Q: TensorVariable) -> TensorVariable:

0 commit comments

Comments
 (0)