fix for collect_raises

tum-ei-eda · May 31, 2024 · 1133353 · 1133353
1 parent 7ad7a9d
commit 1133353
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/seal5/transform/collect_raises/visitor.py b/seal5/transform/collect_raises/visitor.py
@@ -137,7 +137,11 @@ def conditional(self: behav.Conditional, context):
                 stmt = behav.Block([])  # Replace with empty block
                 context.found_raise = False
             else:
-                raise NotImplementedError
+                assert isinstance(stmt, behav.ProcedureCall), "Nesting raises not allowed"
+                assert isinstance(stmt.ref_or_name, arch.Function), "Expected function"
+                assert stmt.ref_or_name.name == "raise", "Expected raise operation"
+                stmt = behav.Block([])  # Replace with empty block
+                context.found_raise = False
             # input("aaaa")
         self.stmts[i] = stmt
         # print("after", context.cond_stack)