Fix bug introduced through merge

spcl · Dec 3, 2024 · d27ec31 · d27ec31
1 parent 3af9a70
commit d27ec31
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/dace/sdfg/state.py b/dace/sdfg/state.py
@@ -2857,13 +2857,9 @@ def all_control_flow_regions(self, recursive=False, load_ext=False,
                             node.load_external(block)
                             yield from node.sdfg.all_control_flow_regions(recursive=recursive, load_ext=load_ext,
                                                                           parent_first=parent_first)
-            elif isinstance(block, ControlFlowRegion):
+            elif isinstance(block, AbstractControlFlowRegion):
                 yield from block.all_control_flow_regions(recursive=recursive, load_ext=load_ext,
                                                           parent_first=parent_first)
-            elif isinstance(block, ConditionalBlock):
-                for _, branch in block.branches:
-                    yield from branch.all_control_flow_regions(recursive=recursive, load_ext=load_ext,
-                                                               parent_first=parent_first)
         if not parent_first:
             yield self