diff --git a/tests/test_runner/test_runner.py b/tests/test_runner/test_runner.py
index 6b984c577e61c20cc03a6af5b42d33729bd5f550..5005927aab3c9b7061c3dbf62dc4e568f1994869 100644
--- a/tests/test_runner/test_runner.py
+++ b/tests/test_runner/test_runner.py
@@ -227,7 +227,8 @@ class CustomRunner(Runner):
                  train_cfg=None,
                  val_cfg=None,
                  test_cfg=None,
-                 optimizer=None,
+                 auto_scale_lr_cfg=None,
+                 optim_wrapper=None,
                  param_scheduler=None,
                  val_evaluator=None,
                  test_evaluator=None,