[BUG] Use event time when no task is cancelled.

erdos-project · Jan 2, 2024 · 746645e · 746645e
1 parent 6ca9935
commit 746645e
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/simulator.py b/simulator.py
@@ -653,6 +653,13 @@ def __create_events_from_task_placement_skip(
         assert (
             not placement.is_placed()
         ), f"Skipping requested for a placed Task {placement.task.unique_name}."
+        self._logger.debug(
+            "[%s] Creating events from the skipping of %s, "
+            "with drop_skipped_tasks set to %s.",
+            time.to(EventTime.Unit.US).time,
+            placement.task.unique_name,
+            drop_skipped_tasks,
+        )
 
         task_events = []
         if drop_skipped_tasks:
@@ -675,13 +682,14 @@ def __create_events_from_task_placement_skip(
             if task_graph.is_cancelled():
                 released_tasks_from_new_task_graph = (
                     self._workload.notify_task_graph_completion(
-                        task_graph, cancelled_task.cancellation_time
+                        task_graph,
+                        time,
                     )
                 )
                 self._logger.info(
                     "[%s] Notified the Workload of the cancellation of %s, "
                     "and received %s new Tasks from new TaskGraphs.",
-                    cancelled_task.cancellation_time.time,
+                    time.to(EventTime.Unit.US).time,
                     task_graph.name,
                     len(released_tasks_from_new_task_graph),
                 )