Small fix in test_cli.py to avoid failure with future version of json…

…argparse (#16156) (cherry picked from commit 50331e0)
Lightning-AI · Dec 21, 2022 · 16f66e1 · 16f66e1
1 parent 18df651
commit 16f66e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tests_pytorch/test_cli.py b/tests/tests_pytorch/test_cli.py
@@ -1233,7 +1233,7 @@ def configure_optimizers(self, optimizer, lr_scheduler=None):
     [optimizer], [scheduler] = cli.model.configure_optimizers()
     assert isinstance(optimizer, SGD)
     assert isinstance(scheduler, StepLR)
-    with mock.patch("sys.argv", ["any.py", "--lr_scheduler=StepLR"]):
+    with mock.patch("sys.argv", ["any.py", "--lr_scheduler=StepLR", "--lr_scheduler.step_size=50"]):
         cli = MyCLI()
     [optimizer], [scheduler] = cli.model.configure_optimizers()
     assert isinstance(optimizer, SGD)