avoid StopIteration in generators

sagemath · Aug 7, 2023 · 4fefac1 · 4fefac1
1 parent 5afb200
commit 4fefac1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/sage/numerical/linear_functions.pyx b/src/sage/numerical/linear_functions.pyx
@@ -1579,7 +1579,10 @@ cdef class LinearConstraint(LinearFunctionOrConstraint):
         while True:
             yield (lhs, rhs)
             lhs = rhs
-            rhs = next(term_iter)
+            try:
+                rhs = next(term_iter)
+            except StopIteration:
+                return
 
     def inequalities(self):
         """
@@ -1612,7 +1615,10 @@ cdef class LinearConstraint(LinearFunctionOrConstraint):
         while True:
             yield (lhs, rhs)
             lhs = rhs
-            rhs = next(term_iter)
+            try:
+                rhs = next(term_iter)
+            except StopIteration:
+                return
 
     def _repr_(self):
         r"""