Skip to content

Commit 96fa8e6

Browse files
committed
Prefer .index over dict, raise nicer exception
1 parent 7ec3112 commit 96fa8e6

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

src/flint/flint_base/flint_base.pyx

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -167,11 +167,10 @@ cdef class flint_mpoly_context(flint_elem):
167167
def variable_to_index(self, var: Union[int, str]):
168168
"""Convert a variable name string or possible index to its index in the context."""
169169
if isinstance(var, str):
170-
vars = {x: i for i, x in enumerate(self.names())}
171-
if var not in vars:
170+
try:
171+
i = self.names().index(var)
172+
except ValueError:
172173
raise ValueError("Variable not in context")
173-
else:
174-
i = vars[var]
175174
elif isinstance(var, int):
176175
if not 0 <= var < self.nvars():
177176
raise IndexError("Generator index out of range")

0 commit comments

Comments
 (0)