From e7ebfbcbb2a977a1f661b54ac70e84d6fedb93a7 Mon Sep 17 00:00:00 2001 From: rasbt Date: Thu, 21 Mar 2024 21:05:03 +0000 Subject: [PATCH] update cli test --- tests/test_cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"]):