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/integrations/test_cut_cross_entropy.py
+++ b/tests/e2e/integrations/test_cut_cross_entropy.py
@@ -4,7 +4,7 @@ Simple end-to-end test for Cut Cross Entropy integration

 import pytest

-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 import get_pytorch_version