Code format

open-mmlab · May 11, 2021 · 858ab1f · 858ab1f
1 parent 8db13b0
commit 858ab1f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_runner/test_hooks.py b/tests/test_runner/test_hooks.py
@@ -21,10 +21,10 @@
 from mmcv.runner import (CheckpointHook, EMAHook, IterTimerHook,
                          MlflowLoggerHook, PaviLoggerHook, WandbLoggerHook,
                          build_runner)
+from mmcv.runner.hooks.hook import HOOKS, Hook
 from mmcv.runner.hooks.lr_updater import (CosineRestartLrUpdaterHook,
                                           OneCycleLrUpdaterHook,
                                           StepLrUpdaterHook)
-from mmcv.runner.hooks.hook import HOOKS, Hook
 
 
 def test_checkpoint_hook():
@@ -124,8 +124,10 @@ def val_step(self, x, optimizer, **kwargs):
 
 
 def test_custom_hook():
+
     @HOOKS.register_module()
     class ToyHook(Hook):
+
         def __init__(self, info, *args, **kwargs):
             super().__init__()
             self.info = info
@@ -857,7 +859,6 @@ def _build_demo_runner(runner_type='EpochBasedRunner',
                        max_iters=None,
                        multi_optimziers=False):
 
-
     log_config = dict(
         interval=1, hooks=[
             dict(type='TextLoggerHook'),