Update formating for yapf

open-mmlab · Jun 28, 2021 · b171407 · b171407
1 parent 27b9fa7
commit b171407
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_runner/test_hooks.py b/tests/test_runner/test_hooks.py
@@ -906,11 +906,13 @@ def test_wandb_hook():
     shutil.rmtree(runner.work_dir)
 
     hook.wandb.init.assert_called_with()
-    hook.wandb.log.assert_called_with({
-        'learning_rate': 0.02,
-        'momentum': 0.95,
-        'global_step': 6
-    }, commit=True)
+    hook.wandb.log.assert_called_with(
+        {
+            'learning_rate': 0.02,
+            'momentum': 0.95,
+            'global_step': 6
+        },
+        commit=True)
     hook.wandb.join.assert_called_with()