summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--sumofsquares/optvariable.py3
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