diff --git a/rpc/service.py b/rpc/service.py index ca069f46..cd25d664 100644 --- a/rpc/service.py +++ b/rpc/service.py @@ -137,6 +137,15 @@ def canonical_task_id(self, stage_id: int): class Servicer(erdos_scheduler_pb2_grpc.SchedulerServiceServicer): def __init__(self) -> None: + # Override some flags + + # Enable orchestrated mode + FLAGS.orchestrated = True + # Set minimum placement push duration to 1s + FLAGS.min_placement_push_duration = 1_000_000 + # Set scheduler runtime to zero + FLAGS.scheduler_runtime = 0 + self._logger = setup_logging( name=__name__, log_dir=FLAGS.log_dir, @@ -251,10 +260,6 @@ async def RegisterFramework(self, request, context): ) self._workload_loader = WorkloadLoader(FLAGS) - # Enable orchestrated mode - FLAGS.orchestrated = True - # Set minimum placement push duration to 1s - FLAGS.min_placement_push_duration = 1_000_000 self._simulator = Simulator( scheduler=self._scheduler, worker_pools=WorkerPools( @@ -583,6 +588,7 @@ async def GetPlacements(self, request, context): for placement in sim_placements: # Ignore virtual placements if placement.task.state < TaskState.RELEASED: + self._logger.debug("[{stime}] Skipping placement: {placement}") continue worker_id = (