From 5981aebfcc6070f064cbda1178beaa4d821c0ecf Mon Sep 17 00:00:00 2001 From: Mauricio Villegas <5780272+mauvilsa@users.noreply.github.com> Date: Fri, 21 Jun 2024 16:38:42 +0200 Subject: [PATCH] Update `test_lightning_cli_help` for future change in jsonargparse (#20002) --- tests/tests_pytorch/test_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tests_pytorch/test_cli.py b/tests/tests_pytorch/test_cli.py index 4c28d6588cea2..8a77fd1440f50 100644 --- 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():