Scheduler: implement remove_condition

kmantel · Dec 5, 2023 · e2abf0c · e2abf0c
1 parent 5e67ca9
commit e2abf0c
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/psyneulink/core/scheduling/scheduler.py b/psyneulink/core/scheduling/scheduler.py
@@ -126,6 +126,21 @@ def _add_condition_set(self, conditions):
         }
         super().add_condition_set(conditions)
 
+    def remove_condition(self, owner_or_condition):
+        try:
+            res = super().remove_condition(owner_or_condition)
+        except AttributeError as e:
+            if "'has no attribute 'remove_condition'" in str(e):
+                raise graph_scheduler.SchedulerError(
+                    f'remove_condition unavailable in your installed graph-scheduler v{graph_scheduler.__version__}'
+                )
+            else:
+                raise
+        else:
+            self.composition._analyze_graph()
+
+        return res
+
     @graph_scheduler.Scheduler.termination_conds.setter
     def termination_conds(self, termination_conds):
         if termination_conds is not None: