Skip to content

Commit 49ed709

Browse files
committed
Update for_helpers.py
1 parent 1b3ace1 commit 49ed709

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

mypyc/irbuild/for_helpers.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1247,16 +1247,13 @@ def get_expr_length_value(
12471247
def _is_supported_forloop_iter(builder: IRBuilder, expr: Expression) -> bool:
12481248
if is_sequence_rprimitive(builder.node_type(expr)):
12491249
return True
1250-
return isinstance(expr, CallExpr) and (
1251-
(
1252-
isinstance(expr.callee, RefExpr)
1253-
and expr.callee.fullname
1254-
in {"builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"}
1255-
)
1256-
or (
1257-
isinstance(expr.callee, MemberExpr) and expr.callee.name in {"keys", "values", "items"}
1258-
)
1259-
)
1250+
if not isinstance(expr, CallExpr):
1251+
return False
1252+
if isinstance(expr.callee, RefExpr):
1253+
return expr.callee.fullname in {"builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"}
1254+
elif isinstance(expr.callee, MemberExpr):
1255+
return expr.callee.fullname in {"keys", "values", "items"}
1256+
return False
12601257

12611258

12621259
def _create_iterable_lexpr(index_name: str, index_type: Type) -> NameExpr:

0 commit comments

Comments
 (0)