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 4b25ec6..cfb6bea 100644 --- a/polymatrix/expression/mixins/divisionexprmixin.py +++ b/polymatrix/expression/mixins/divisionexprmixin.py @@ -30,8 +30,8 @@ class DivisionExprMixin(ExpressionBaseMixin): state: ExpressionState, ) -> tuple[ExpressionState, PolyMatrix]: - if self in state.cached_polymatrix: - return state, state.cached_polymatrix[self] + if self in state.cache: + return state, state.cache[self] state, left = self.left.apply(state=state) state, right = self.right.apply(state=state) @@ -76,7 +76,7 @@ class DivisionExprMixin(ExpressionBaseMixin): state = dataclasses.replace( state, auxillary_equations=state.auxillary_equations | {division_variable: auxillary_terms}, - cached_polymatrix=state.cached_polymatrix | {self: poly_matrix}, + cached_polymatrix=state.cache | {self: poly_matrix}, ) return state, poly_matrix |