Fix

2025-01-10 17:32:33 +00:00
parent 210c58a4db
commit c9e37496cb
26 changed files with 26 additions and 26 deletions
--- a/tests/e2e/test_llama.py
+++ b/tests/e2e/test_llama.py
@@ -7,7 +7,7 @@ import os

 from e2e.utils import check_model_output_exists

-from axolotl.common.cli import TrainerCliArgs
+from axolotl.cli.args import TrainerCliArgs
 from axolotl.common.datasets import load_datasets
 from axolotl.train import train
 from axolotl.utils.config import normalize_config