[dace] Fix formatting

GridTools · Jan 31, 2024 · 43346b3 · 43346b3
1 parent c35d5dc
commit 43346b3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py
@@ -1192,9 +1192,11 @@ def _visit_reduce(self, node: itir.FunCall):
             )
 
             input_mapping = {
-                param: dace.Memlet(data=arg.value.data, subset="0")
-                if arg.value.desc(self.context.body).shape == (1,)
-                else create_memlet_at(arg.value.data, nreduce_index)
+                param: (
+                    dace.Memlet(data=arg.value.data, subset="0")
+                    if arg.value.desc(self.context.body).shape == (1,)
+                    else create_memlet_at(arg.value.data, nreduce_index)
+                )
                 for (param, _), arg in zip(inner_inputs, args)
             }
             output_mapping = {