Add type hint

open-mmlab · May 25, 2022 · 206f04c · 206f04c
1 parent 05bdd16
commit 206f04c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mmcv/runner/builder.py b/mmcv/runner/builder.py
@@ -1,17 +1,18 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import copy
+from typing import Optional
 
 from ..utils import Registry
 
 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')