diff --git a/tests/test_cli.py b/tests/test_cli.py index 85510d9ae1..2c994fcf96 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -20,10 +20,10 @@ def test_cli(): """Available subcommands: download Download weights or tokenizer data from the Hugging Face Hub. - chat Chat with a model. - evaluate Evaluate a model with the LM Evaluation Harness.""" + chat Chat with a model.""" in out ) + assert ("""evaluate Evaluate a model with the LM Evaluation Harness.""") in out out = StringIO() with pytest.raises(SystemExit), redirect_stdout(out), mock.patch("sys.argv", ["litgpt", "finetune", "-h"]):