From 6dd47edcb8fe03b9dfc8d6430ecc52f073bdf072 Mon Sep 17 00:00:00 2001
From: Dan Saunders <danjsaund@gmail.com>
Date: Fri, 10 Jan 2025 18:24:16 +0000
Subject: [PATCH] fire CLI fixes

---
 tests/cli/test_cli_base.py      | 4 +---
 tests/cli/test_cli_interface.py | 1 +
 tests/cli/test_cli_shard.py     | 4 +---
 3 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/tests/cli/test_cli_base.py b/tests/cli/test_cli_base.py
index 6dbae045f..f8f1edfa3 100644
--- 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
diff --git a/tests/cli/test_cli_interface.py b/tests/cli/test_cli_interface.py
index 8b5fec17f..935fb85b8 100644
--- a/tests/cli/test_cli_interface.py
+++ b/tests/cli/test_cli_interface.py
@@ -23,6 +23,7 @@ def test_build_command():
         "--batch-size",
         "8",
         "--debug",
+        "--nouse-fp16",
     ]
 
 
diff --git a/tests/cli/test_cli_shard.py b/tests/cli/test_cli_shard.py
index 3176ed27e..86766e17f 100644
--- a/tests/cli/test_cli_shard.py
+++ b/tests/cli/test_cli_shard.py
@@ -12,14 +12,12 @@ def test_shard_with_accelerate(cli_runner, config_path):
         result = cli_runner.invoke(cli, ["shard", str(config_path), "--accelerate"])
 
         assert mock.called
-        assert mock.call_args.args[0] == [
+        assert mock.call_args.args[0][:5] == [
             "accelerate",
             "launch",
             "-m",
             "axolotl.cli.shard",
             str(config_path),
-            "--debug-num-examples",
-            "0",
         ]
         assert mock.call_args.kwargs == {"check": True}
         assert result.exit_code == 0