diff options
| -rw-r--r-- | polymatrix/expression/mixins/parametrizeexprmixin.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/polymatrix/expression/mixins/parametrizeexprmixin.py b/polymatrix/expression/mixins/parametrizeexprmixin.py index 0676cea..644198e 100644 --- a/polymatrix/expression/mixins/parametrizeexprmixin.py +++ b/polymatrix/expression/mixins/parametrizeexprmixin.py @@ -43,7 +43,7 @@ class ParametrizeExprMixin(ExpressionBaseMixin): # FIXME: not sure this behaviour is intuitive, discuss if v := state.get_variable_from_name_or(self.name, if_not_present=False): - start, end = state.offset_dict[self.name] + start, end = state.offset_dict[v] if (nrows * ncols) != (end - start): raise ValueError("Cannot parametrize {self.underlying} with variable {v} " "found in state object, because its shape {(nrow, ncols)} " |