formatting done

spcl · Oct 30, 2023 · fdc68d0 · fdc68d0
1 parent f66ef87
commit fdc68d0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/dace/transformation/dataflow/wcr_conversion.py b/dace/transformation/dataflow/wcr_conversion.py
@@ -11,6 +11,7 @@
 from dace.transformation import helpers
 from dace import propagate_memlet
 
+
 class AugAssignToWCR(transformation.SingleStateTransformation):
     """
     Converts an augmented assignment ("a += b", "a = a + b") into a tasklet
@@ -154,9 +155,7 @@ def apply(self, state: SDFGState, sdfg: SDFG):
 
         # If state fission is necessary to keep semantics, do it first
         if state.in_degree(input) > 0:
-            subgraph_nodes = set(
-                [e.src for e in state.bfs_edges(input, reverse=True)]
-            )
+            subgraph_nodes = set([e.src for e in state.bfs_edges(input, reverse=True)])
             subgraph_nodes.add(input)
 
             subgraph = StateSubgraphView(state, subgraph_nodes)