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

…argparse (#16156)
Lightning-AI · Dec 21, 2022 · 50331e0 · 50331e0
1 parent 0630444
commit 50331e0
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
@@ -1314,7 +1314,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)