From 6cc89611337bb74555fd16b4be78681fb7e36573 Mon Sep 17 00:00:00 2001 From: Xingyao Wang Date: Tue, 30 Apr 2024 15:09:22 +0800 Subject: [PATCH] fix arg parser that's broken for some reason --- tests/unit/test_arg_parser.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/unit/test_arg_parser.py b/tests/unit/test_arg_parser.py index 73e8e811b6..d815f3fa46 100644 --- a/tests/unit/test_arg_parser.py +++ b/tests/unit/test_arg_parser.py @@ -9,8 +9,8 @@ def test_help_message(capsys): parser.parse_args(['--help']) captured = capsys.readouterr() expected_help_message = """ -usage: pytest [-h] [-d DIRECTORY] [-t TASK] [-f FILE] [-c AGENT_CLS] -[-m MODEL_NAME] [-i MAX_ITERATIONS] [-n MAX_CHARS] +usage: pytest [-h] [-d DIRECTORY] [-t TASK] [-f FILE] [-c AGENT_CLS] [-m MODEL_NAME] [-i MAX_ITERATIONS] + [-n MAX_CHARS] Run an agent with a specific task @@ -19,8 +19,7 @@ options: -d DIRECTORY, --directory DIRECTORY The working directory for the agent -t TASK, --task TASK The task for the agent to perform - -f FILE, --file FILE Path to a file containing the task. Overrides -t if - both are provided. + -f FILE, --file FILE Path to a file containing the task. Overrides -t if both are provided. -c AGENT_CLS, --agent-cls AGENT_CLS The agent class to use -m MODEL_NAME, --model-name MODEL_NAME @@ -28,8 +27,7 @@ options: -i MAX_ITERATIONS, --max-iterations MAX_ITERATIONS The maximum number of iterations to run the agent -n MAX_CHARS, --max-chars MAX_CHARS - The maximum number of characters to send to and - receive from LLM per task + The maximum number of characters to send to and receive from LLM per task """ actual_lines = captured.out.strip().split('\n') expected_lines = expected_help_message.strip().split('\n')