[BUG] Do not generate ChooseExpressions until the plan-ahead no consi…

…deration gap ends.
erdos-project · Apr 7, 2024 · c6bed94 · c6bed94
1 parent 5e9c160
commit c6bed94
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/schedulers/tetrisched_scheduler.py b/schedulers/tetrisched_scheduler.py
@@ -1015,7 +1015,11 @@ def _get_time_discretizations_until(
         """
         time_discretization = self._time_discretization.to(EventTime.Unit.US).time
         start_time = (
-            current_time.to(EventTime.Unit.US).time // time_discretization
+            (
+                current_time.to(EventTime.Unit.US).time
+                + self._plan_ahead_no_consideration_gap.to(EventTime.Unit.US).time
+            )
+            // time_discretization
         ) * time_discretization
         end_time = end_time.to(EventTime.Unit.US).time