Fix format

GridTools · Sep 17, 2024 · 77c900b · 77c900b
1 parent 8f5f7dc
commit 77c900b
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/gt4py/next/iterator/transforms/infer_domain.py b/src/gt4py/next/iterator/transforms/infer_domain.py
@@ -11,8 +11,8 @@
 from gt4py.next.common import Dimension
 from gt4py.next.iterator import ir as itir
 from gt4py.next.iterator.ir_utils import common_pattern_matcher as cpm, ir_makers as im
-from gt4py.next.iterator.transforms.global_tmps import AUTO_DOMAIN, SymbolicDomain, domain_union
 from gt4py.next.iterator.transforms import trace_shifts
+from gt4py.next.iterator.transforms.global_tmps import AUTO_DOMAIN, SymbolicDomain, domain_union
 
 
 def _merge_domains(
@@ -27,17 +27,15 @@ def _merge_domains(
 
     return new_domains
 
+
 def extract_shifts_and_translate_domains(
     stencil: itir.Expr,
     input_ids: list[str],
     target_domain: SymbolicDomain,
     offset_provider: Dict[str, Dimension],
     accessed_domains: Dict[str, SymbolicDomain],
 ):
-    shifts_results = trace_shifts.trace_stencil(
-        stencil,
-        num_args=len(input_ids)
-    )
+    shifts_results = trace_shifts.trace_stencil(stencil, num_args=len(input_ids))
 
     for in_field_id, shifts_list in zip(input_ids, shifts_results, strict=True):
         new_domains = [