fix

tests/e2e/patched/test_4d_multipack_llama.py

@@ -6,7 +6,7 @@ import logging
 import os
 import unittest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_cli_integrations.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 import yaml
 
-from axolotl.cli import load_cfg
+from axolotl.cli.config import load_cfg
 from axolotl.utils.dict import DictDefault
 
 

tests/e2e/patched/test_fa_xentropy.py

@@ -8,7 +8,7 @@ import os
 import pytest
 from transformers.utils import is_torch_bf16_gpu_available
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_falcon_samplepack.py

@@ -6,7 +6,7 @@ import logging
 import os
 import unittest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_fused_llama.py

@@ -9,7 +9,7 @@ import unittest
 import pytest
 from transformers.utils import is_torch_bf16_gpu_available
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_llama_s2_attention.py

@@ -8,7 +8,7 @@ import unittest
 
 import pytest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_lora_llama_multipack.py

@@ -9,7 +9,7 @@ import unittest
 import pytest
 from transformers.utils import is_auto_gptq_available, is_torch_bf16_gpu_available
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_mistral_samplepack.py

@@ -6,7 +6,7 @@ import logging
 import os
 import unittest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_mixtral_samplepack.py

@@ -6,7 +6,7 @@ import logging
 import os
 import unittest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_phi_multipack.py

@@ -6,7 +6,7 @@ import logging
 import os
 import unittest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_resume.py

@@ -9,7 +9,7 @@ import subprocess
 
 from transformers.utils import is_torch_bf16_gpu_available
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/patched/test_unsloth_qlora.py

@@ -6,7 +6,7 @@ import os
 
 import pytest
 
-from axolotl.cli import load_datasets
+from axolotl.cli.datasets import load_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config