[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

mypyc/irbuild/for_helpers.py

@@ -1225,17 +1225,18 @@ def get_expr_length_value(
 def _is_supported_forloop_iter(builder: IRBuilder, expr: Expression) -> bool:
     if is_sequence_rprimitive(builder.node_type(expr)):
         return True
-    return (
-        isinstance(expr, CallExpr)
-        and (
-            (isinstance(expr.callee, RefExpr) and expr.callee.fullname in {
-                "builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"
-            })
-            or (isinstance(expr.callee, MemberExpr) and expr.callee.name in {"keys", "values", "items"})
+    return isinstance(expr, CallExpr) and (
+        (
+            isinstance(expr.callee, RefExpr)
+            and expr.callee.fullname
+            in {"builtins.range", "builtins.enumerate", "builtins.zip", "builtins.reversed"}
+        )
+        or (
+            isinstance(expr.callee, MemberExpr) and expr.callee.name in {"keys", "values", "items"}
         )
     )
 
-    
+
 def _create_iterable_lexpr(index_name: str, index_type: Type) -> NameExpr:
     """This helper spoofs a NameExpr to use as the lvalue in one of the for loop helpers."""
     index = NameExpr(index_name)

mypyc/irbuild/specialize.py

@@ -77,8 +77,8 @@
 )
 from mypyc.irbuild.builder import IRBuilder
 from mypyc.irbuild.for_helpers import (
-    _is_supported_forloop_iter,
     _create_iterable_lexpr,
+    _is_supported_forloop_iter,
     comprehension_helper,
     for_loop_helper,
     sequence_from_generator_preallocate_helper,
@@ -435,15 +435,7 @@ def body_insts() -> None:
             index = _create_iterable_lexpr(index_name, index_type)
             index_reg = builder.add_local_reg(index.node, builder.type_to_rtype(index_type))
 
-            for_loop_helper(
-                builder,
-                index,
-                arg,
-                body_insts,
-                None,
-                is_async=False,
-                line=expr.line,
-            )
+            for_loop_helper(builder, index, arg, body_insts, None, is_async=False, line=expr.line)
             builder.goto_and_activate(loop_exit)
             return retval
     return None
@@ -476,20 +468,12 @@ def body_insts() -> None:
                 builder.assign(retval, builder.false(), -1)
                 builder.goto(loop_exit)
                 builder.activate_block(true_block)
-            
+
             index_type = builder._analyze_iterable_item_type(arg)
             index = _create_iterable_lexpr(index_name, index_type)
             index_reg = builder.add_local_reg(index.node, builder.type_to_rtype(index_type))
 
-            for_loop_helper(
-                builder,
-                index,
-                arg,
-                body_insts,
-                None,
-                is_async=False,
-                line=expr.line,
-            )
+            for_loop_helper(builder, index, arg, body_insts, None, is_async=False, line=expr.line)
             builder.goto_and_activate(loop_exit)
             return retval
     return None