Skip to content

Commit 2b43f62

Browse files
Update specialize.py
1 parent df669fe commit 2b43f62

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

mypyc/irbuild/specialize.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
RefExpr,
3131
StrExpr,
3232
SuperExpr,
33+
SymbolNode,
3334
TupleExpr,
3435
Var,
3536
)
@@ -436,7 +437,10 @@ def body_insts() -> None:
436437

437438
index_type = builder._analyze_iterable_item_type(arg)
438439
index = create_synthetic_nameexpr(builder, index_name, index_type)
439-
index_reg = builder.add_local_reg(index.node, builder.type_to_rtype(index_type))
440+
index_reg = builder.add_local_reg(
441+
cast(SymbolNode, index.node),
442+
builder.type_to_rtype(index_type),
443+
)
440444

441445
for_loop_helper(builder, index, arg, body_insts, None, is_async=False, line=expr.line)
442446
builder.goto_and_activate(loop_exit)
@@ -474,7 +478,10 @@ def body_insts() -> None:
474478

475479
index_type = builder._analyze_iterable_item_type(arg)
476480
index = create_synthetic_nameexpr(builder, index_name, index_type)
477-
index_reg = builder.add_local_reg(index.node, builder.type_to_rtype(index_type))
481+
index_reg = builder.add_local_reg(
482+
cast(SymbolNode, index.node),
483+
builder.type_to_rtype(index_type),
484+
)
478485

479486
for_loop_helper(builder, index, arg, body_insts, None, is_async=False, line=expr.line)
480487
builder.goto_and_activate(loop_exit)

0 commit comments

Comments
 (0)