[bug] Increasing workers size - missing variables during initialization

In the BlockAllocationTaskScheduler the workers are initialized with the following additional parameters:

{
    "worker_id": worker_id,
    "stop_function": lambda: _interrupt_bootup_dict[self_id],
    "bootup_event": bootup_events[worker_id],
    "next_bootup_event": (
        bootup_events[worker_id + 1]
        if worker_id + 1 < self._max_workers
        else None
    ),
    "alive_workers": alive_workers,
    "alive_workers_lock": alive_workers_lock,
}

But these additional parameters are not provided when the workers are rescaled using self.max_workers.

Locate the BlockAllocationTaskScheduler implementation and compare its initial worker setup with the rescaling path triggered by self.max_workers. Trace how the listed worker parameters are constructed and passed in each path. Done means workers created during rescaling receive the same required initialization context and the existing scheduler behavior remains intact.