Skip to content

Commit b0edaeb

Browse files
Update for_helpers.py
1 parent 02d4739 commit b0edaeb

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
@@ -1225,16 +1225,13 @@ def get_expr_length_value(
12251225
def _is_supported_forloop_iter(builder: IRBuilder, expr: Expression) -> bool:
12261226
if is_sequence_rprimitive(builder.node_type(expr)):
12271227
return True
1228-
return isinstance(expr, CallExpr) and (
1229-
(
1230-
isinstance(expr.callee, RefExpr)
1231-
and expr.callee.fullname
1232-
in {"builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"}
1233-
)
1234-
or (
1235-
isinstance(expr.callee, MemberExpr) and expr.callee.name in {"keys", "values", "items"}
1236-
)
1237-
)
1228+
if not isinstance(expr, CallExpr):
1229+
return False
1230+
if isinstance(expr.callee, RefExpr):
1231+
return expr.callee.fullname in {"builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"}
1232+
elif isinstance(expr.callee, MemberExpr):
1233+
return expr.callee.fullname in {"keys", "values", "items"}
1234+
return False
12381235

12391236

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

0 commit comments

Comments
 (0)