Unverified Commit 84a544fb authored by Yvette Zhao's avatar Yvette Zhao Committed by GitHub
Browse files

Add type hints for runner/builder (#1985)

parent ef2b927b
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import copy import copy
from typing import Optional
from ..utils import Registry from ..utils import Registry
...@@ -7,11 +8,11 @@ RUNNERS = Registry('runner') ...@@ -7,11 +8,11 @@ RUNNERS = Registry('runner')
RUNNER_BUILDERS = Registry('runner builder') RUNNER_BUILDERS = Registry('runner builder')
def build_runner_constructor(cfg): def build_runner_constructor(cfg: dict):
return RUNNER_BUILDERS.build(cfg) 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) runner_cfg = copy.deepcopy(cfg)
constructor_type = runner_cfg.pop('constructor', constructor_type = runner_cfg.pop('constructor',
'DefaultRunnerConstructor') 'DefaultRunnerConstructor')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment