File size: 1,087 Bytes
7accb91 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import logging
from hydra.utils import instantiate
from omegaconf import DictConfig
from nuplan.planning.script.builders.utils.utils_type import is_target_type, validate_type
from nuplan.planning.utils.multithreading.worker_pool import WorkerPool
from nuplan.planning.utils.multithreading.worker_parallel import SingleMachineParallelExecutor
from nuplan.planning.utils.multithreading.worker_sequential import Sequential
logger = logging.getLogger(__name__)
def build_worker(cfg: DictConfig) -> WorkerPool:
"""
Builds the worker.
:param cfg: DictConfig. Configuration that is used to run the experiment.
:return: Instance of WorkerPool.
"""
logger.info('Building WorkerPool...')
worker: WorkerPool = (
instantiate(cfg.worker)
if (
is_target_type(cfg.worker, SingleMachineParallelExecutor)
or is_target_type(cfg.worker, Sequential)
)
else instantiate(cfg.worker, output_dir=cfg.output_dir)
)
validate_type(worker, WorkerPool)
logger.info('Building WorkerPool...DONE!')
return worker
|