fix tests from merge conflict

axolotl-ai-cloud · Jan 29, 2025 · 6551682 · 6551682
1 parent 165c530
commit 6551682
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/e2e/test_deepseekv3.py b/tests/e2e/test_deepseekv3.py
@@ -67,7 +67,7 @@ def test_lora_deepseekv3(self, temp_dir, sample_packing):
         cli_args = TrainerCliArgs()
         dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args)
 
-        train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta)
+        train(cfg=cfg, dataset_meta=dataset_meta)
         assert (Path(temp_dir) / "adapter_model.safetensors").exists()
 
     @pytest.mark.parametrize(
@@ -109,5 +109,5 @@ def test_fft_deepseekv3(self, temp_dir, sample_packing):
         cli_args = TrainerCliArgs()
         dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args)
 
-        train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta)
+        train(cfg=cfg, dataset_meta=dataset_meta)
         assert (Path(temp_dir) / "model.safetensors").exists()