simplify visit_For

tserg · Sep 28, 2023 · 9880fbf · 9880fbf
1 parent c73080e
commit 9880fbf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/vyper/semantics/analysis/local.py b/vyper/semantics/analysis/local.py
@@ -360,7 +360,7 @@ def visit_For(self, node):
                 if bound is None:
                     n_val = derive_folded_value(n)
                     if not isinstance(n_val, int):
-                        raise StateAccessViolation("Value must be a literal", n)
+                        raise StateAccessViolation("Value must be a literal integer", n)
                     if n_val <= 0:
                         raise StructureException("For loop must have at least 1 iteration", args[0])
                     type_list = get_possible_types_from_node(n)
@@ -398,9 +398,7 @@ def visit_For(self, node):
                         )
 
                     right_val = derive_folded_value(args[1].right)
-                    if not isinstance(args[1].right, vy_ast.Int) and not (
-                        isinstance(args[1].right, vy_ast.Name) and right_val
-                    ):
+                    if right_val is None:
                         raise InvalidLiteral("Literal must be an integer", args[1].right)
                     if right_val < 1:
                         raise StructureException(