Update index validation

dwavesystems · Sep 28, 2024 · d2c94b2 · d2c94b2
1 parent af58ab3
commit d2c94b2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dwave/optimization/model.pyx b/dwave/optimization/model.pyx
@@ -254,7 +254,8 @@ cdef class Model:
         Returns:
             Feasibility of the state.
         """
-        cdef int num_states = self.state_size()
+        cdef Py_ssize_t num_states = self.states.size()
+
         if not -num_states <= index < num_states:
             raise ValueError(f"index out of range: {index}")
         elif index < 0:  # allow negative indexing