Scheduler: get_clock: use default_execution_id if not passed in (#111)

kmantel · Dec 11, 2023 · a99ec73 · a99ec73
1 parent 6485a3e
commit a99ec73
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/graph_scheduler/scheduler.py b/src/graph_scheduler/scheduler.py
@@ -901,7 +901,10 @@ def get_absolute_conditions(self, termination_conds=None):
             if cond.is_absolute
         }
 
-    def get_clock(self, execution_id):
+    def get_clock(self, execution_id=NotImplemented):
+        if execution_id is NotImplemented:
+            execution_id = self.default_execution_id
+
         try:
             return self.clocks[execution_id.default_execution_id]
         except AttributeError: