Update test_lightning_cli_help for future change in jsonargparse (#…

…20002)
Lightning-AI · Jun 21, 2024 · 5981aeb · 5981aeb
1 parent d3a0ada
commit 5981aeb
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
@@ -466,7 +466,7 @@ def test_lightning_cli_help():
     ), pytest.raises(SystemExit):
         any_model_any_data_cli()
 
-    assert "--data.init_args.data_dir" in out.getvalue()
+    assert ("--data.data_dir" in out.getvalue()) or ("--data.init_args.data_dir" in out.getvalue())
 
 
 def test_lightning_cli_print_config():