Skip to content

Commit

Permalink
rename some variables
Browse files Browse the repository at this point in the history
  • Loading branch information
charles-cooper committed Jan 7, 2024
1 parent 2351bb2 commit 6e04ed5
Showing 1 changed file with 9 additions and 8 deletions.
17 changes: 9 additions & 8 deletions vyper/codegen/stmt.py
Original file line number Diff line number Diff line change
Expand Up @@ -232,23 +232,24 @@ def parse_For(self):

def _parse_For_range(self):
assert "type" in self.stmt.target.target._metadata
iter_typ = self.stmt.target.target._metadata["type"]
target_type = self.stmt.target.target._metadata["type"]

# Get arg0
for_iter: vy_ast.Call = self.stmt.iter
args_len = len(for_iter.args)
range_call: vy_ast.Call = self.stmt.iter
assert isinstance(range_call, vy_ast.Call)
args_len = len(range_call.args)
if args_len == 1:
arg0, arg1 = (IRnode.from_list(0, typ=iter_typ), for_iter.args[0])
arg0, arg1 = (IRnode.from_list(0, typ=target_type), range_call.args[0])
elif args_len == 2:
arg0, arg1 = for_iter.args
arg0, arg1 = range_call.args
else: # pragma: nocover
raise TypeCheckFailure("unreachable: bad # of arguments to range()")

with self.context.range_scope():
start = Expr.parse_value_expr(arg0, self.context)
end = Expr.parse_value_expr(arg1, self.context)
kwargs = {
s.arg: Expr.parse_value_expr(s.value, self.context) for s in for_iter.keywords
s.arg: Expr.parse_value_expr(s.value, self.context) for s in range_call.keywords
}

if "bound" in kwargs:
Expand All @@ -268,8 +269,8 @@ def _parse_For_range(self):
raise TypeCheckFailure("unreachable: unchecked 0 bound")

varname = self.stmt.target.target.id
i = IRnode.from_list(self.context.fresh_varname("range_ix"), typ=iter_typ)
iptr = self.context.new_variable(varname, iter_typ)
i = IRnode.from_list(self.context.fresh_varname("range_ix"), typ=target_type)
iptr = self.context.new_variable(varname, target_type)

self.context.forvars[varname] = True

Expand Down

0 comments on commit 6e04ed5

Please sign in to comment.