Skip to content

Commit 7dd362a

Browse files
Update for_helpers.py
1 parent e4835ae commit 7dd362a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

mypyc/irbuild/for_helpers.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -442,17 +442,17 @@ def make_for_loop_generator(
442442
target_type = builder.get_sequence_type(expr)
443443
for_list = ForSequence(builder, index, body_block, loop_exit, line, nested)
444444

445-
if isinstance(expr, IndexExpr) and all(
445+
if isinstance(expr, IndexExpr) and isinstance(expr.index, SliceExpr) and all(
446446
s is None or isinstance(constant_fold_expr(builder, s), int)
447-
for s in (expr.start, expr.stop, expr.step)
447+
for s in (expr.index.start, expr.index.stop, expr.index.step)
448448
):
449449
for_list.init(
450-
builder.accept(expr.expr),
450+
builder.accept(expr.base),
451451
target_type,
452452
reverse=False,
453-
start=expr.start,
454-
stop=expr.stop,
455-
step=expr.step,
453+
start=expr.index.start,
454+
stop=expr.index.stop,
455+
step=expr.index.step,
456456
)
457457
else:
458458
for_list.init(builder.accept(expr), target_type, reverse=False)

0 commit comments

Comments
 (0)