diff --git a/mmcv/runner/builder.py b/mmcv/runner/builder.py index 77c96ba0b..008da32aa 100644 --- a/mmcv/runner/builder.py +++ b/mmcv/runner/builder.py @@ -1,5 +1,6 @@ # Copyright (c) OpenMMLab. All rights reserved. import copy +from typing import Optional from ..utils import Registry @@ -7,11 +8,11 @@ RUNNERS = Registry('runner') RUNNER_BUILDERS = Registry('runner builder') -def build_runner_constructor(cfg): +def build_runner_constructor(cfg: dict): return RUNNER_BUILDERS.build(cfg) -def build_runner(cfg, default_args=None): +def build_runner(cfg: dict, default_args: Optional[dict] = None): runner_cfg = copy.deepcopy(cfg) constructor_type = runner_cfg.pop('constructor', 'DefaultRunnerConstructor')