tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fire CLI fixes

Browse Source
This commit is contained in:
Dan Saunders
2025-01-10 18:24:16 +00:00
parent 7aca08ff60
commit 6dd47edcb8
3 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/cli/test_cli_base.py
View File

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

1
tests/cli/test_cli_interface.py
View File

@@ -23,6 +23,7 @@ def test_build_command():
"--batch-size", "--batch-size",
"8", "8",
"--debug", "--debug",
"--nouse-fp16",
] ]

4
tests/cli/test_cli_shard.py
View File

@@ -12,14 +12,12 @@ def test_shard_with_accelerate(cli_runner, config_path):
result = cli_runner.invoke(cli, ["shard", str(config_path), "--accelerate"]) result = cli_runner.invoke(cli, ["shard", str(config_path), "--accelerate"])
assert mock.called assert mock.called
assert mock.call_args.args[0] == [ assert mock.call_args.args[0][:5] == [
"accelerate", "accelerate",
"launch", "launch",
"-m", "-m",
"axolotl.cli.shard", "axolotl.cli.shard",
str(config_path), str(config_path),
"--debug-num-examples",
"0",
] ]
assert mock.call_args.kwargs == {"check": True} assert mock.call_args.kwargs == {"check": True}
assert result.exit_code == 0 assert result.exit_code == 0