fire CLI fixes

2025-01-10 18:24:16 +00:00
parent 7aca08ff60
commit 6dd47edcb8
3 changed files with 3 additions and 6 deletions
--- a/tests/cli/test_cli_base.py
+++ b/tests/cli/test_cli_base.py
@@ -43,14 +43,12 @@ class BaseCliTest:
            result = cli_runner.invoke(cli, [command, str(config_path)])

            assert mock.called
-            assert mock.call_args.args[0] == [
+            assert mock.call_args.args[0][:5] == [
                "accelerate",
                "launch",
                "-m",
                f"axolotl.cli.{command}",
                str(config_path),
-                "--debug-num-examples",
-                "0",
            ]
            assert mock.call_args.kwargs == {"check": True}
            assert result.exit_code == 0