diff options
| -rw-r--r-- | sumofsquares/optvariable.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/sumofsquares/optvariable.py b/sumofsquares/optvariable.py index 5ee7cbc..6e933b6 100644 --- a/sumofsquares/optvariable.py +++ b/sumofsquares/optvariable.py @@ -15,7 +15,8 @@ class OptVariableMixin(ExpressionBaseMixin, Variable): """ Optimization (decision) variable. """ @override def apply(self, state: ExpressionState) -> tuple[ExpressionState, PolyMatrix]: - state, indices = state.index(self) + state = state.register(self) + indices = state.get_indices(self) p = PolyMatrixDict() rows, cols = self.shape |