From 24487bd6466298a66276c5b044011c752fa9aa69 Mon Sep 17 00:00:00 2001 From: Nao Pross Date: Sat, 4 May 2024 19:11:26 +0200 Subject: Fix regression in polymatrix.to_sympy --- polymatrix/expression/to.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/polymatrix/expression/to.py b/polymatrix/expression/to.py index 2a43c0e..2707fb4 100644 --- a/polymatrix/expression/to.py +++ b/polymatrix/expression/to.py @@ -49,7 +49,8 @@ def to_sympy( expr: Expression, ) -> StateMonadMixin[ExpressionState, sympy.Expr | sympy.Matrix]: - def polymatrix_to_sympy(state: ExpressionState) -> sympy.Expr | sympy.Matrix: + def polymatrix_to_sympy(state: ExpressionState) -> tuple[ExpressionState, sympy.Expr | sympy.Matrix]: + # Convert to polymatrix state, pm = expr.apply(state) @@ -72,7 +73,7 @@ def to_sympy( if math.prod(pm.shape) == 1: # just return the expression - return m[0, 0] + return state, m[0, 0] - return m + return state, m return init_state_monad(polymatrix_to_sympy) -- cgit v1.2.1