diff options
Diffstat (limited to 'polymatrix/expression/mixins/divisionexprmixin.py')
| -rw-r--r-- | polymatrix/expression/mixins/divisionexprmixin.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/polymatrix/expression/mixins/divisionexprmixin.py b/polymatrix/expression/mixins/divisionexprmixin.py index de5f685..7b5e0c5 100644 --- a/polymatrix/expression/mixins/divisionexprmixin.py +++ b/polymatrix/expression/mixins/divisionexprmixin.py @@ -5,7 +5,7 @@ import dataclasses from polymatrix.expression.init.initpolymatrix import init_poly_matrix from polymatrix.expression.mixins.expressionbasemixin import ExpressionBaseMixin from polymatrix.expression.polymatrix import PolyMatrix -from polymatrix.polymatrixexprstate import PolyMatrixExprState +from polymatrix.expression.expressionstate import ExpressionState class DivisionExprMixin(ExpressionBaseMixin): @@ -27,8 +27,8 @@ class DivisionExprMixin(ExpressionBaseMixin): # overwrites abstract method of `ExpressionBaseMixin` def apply( self, - state: PolyMatrixExprState, - ) -> tuple[PolyMatrixExprState, PolyMatrix]: + state: ExpressionState, + ) -> tuple[ExpressionState, PolyMatrix]: if self in state.cached_polymatrix: return state, state.cached_polymatrix[self] |