Formatting

GridTools · Apr 19, 2024 · d42fa12 · d42fa12
1 parent 873c60e
commit d42fa12
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/gt4py/next/iterator/type_system/inference.py b/src/gt4py/next/iterator/type_system/inference.py
@@ -78,9 +78,13 @@ def _is_compatible_type(type_a: ts.TypeSpec, type_b: ts.TypeSpec):
     elif isinstance(type_a, ts.FunctionType):
         for arg_a, arg_b in zip(type_a.pos_only_args, type_b.pos_only_args, strict=True):
             is_compatible &= _is_compatible_type(arg_a, arg_b)
-        for arg_a, arg_b in zip(type_a.pos_or_kw_args.values(), type_b.pos_or_kw_args.values(), strict=True):
+        for arg_a, arg_b in zip(
+            type_a.pos_or_kw_args.values(), type_b.pos_or_kw_args.values(), strict=True
+        ):
             is_compatible &= _is_compatible_type(arg_a, arg_b)
-        for arg_a, arg_b in zip(type_a.kw_only_args.values(), type_b.kw_only_args.values(), strict=True):
+        for arg_a, arg_b in zip(
+            type_a.kw_only_args.values(), type_b.kw_only_args.values(), strict=True
+        ):
             is_compatible &= _is_compatible_type(arg_a, arg_b)
         is_compatible &= _is_compatible_type(type_a.returns, type_b.returns)
     else:

diff --git a/tests/next_tests/unit_tests/iterator_tests/transforms_tests/test_global_tmps.py b/tests/next_tests/unit_tests/iterator_tests/transforms_tests/test_global_tmps.py
@@ -390,7 +390,15 @@ def test_collect_tmps_info():
         fencil=ir.FencilDefinition(
             id="f",
             function_definitions=[],
-            params=[i, j, k, inp, out, im.sym("_gtmp_0", i_field_type), im.sym("_gtmp_1", i_field_type)],
+            params=[
+                i,
+                j,
+                k,
+                inp,
+                out,
+                im.sym("_gtmp_0", i_field_type),
+                im.sym("_gtmp_1", i_field_type),
+            ],
             closures=[
                 ir.StencilClosure(
                     domain=tmp_domain,