fix

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

tests/e2e/integrations/test_liger.py

@@ -4,7 +4,7 @@ Simple end-to-end test for Liger integration
 
 from e2e.utils import require_torch_2_4_1
 
-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, prepare_plugins

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

tests/e2e/test_dpo.py

@@ -9,7 +9,7 @@ from pathlib import Path
 
 import pytest
 
-from axolotl.cli import load_rl_datasets
+from axolotl.cli.datasets import load_rl_datasets
 from axolotl.common.cli import TrainerCliArgs
 from axolotl.train import train
 from axolotl.utils.config import normalize_config

tests/e2e/test_embeddings_lr.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/test_falcon.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/test_llama.py

@@ -7,7 +7,7 @@ import os
 
 from e2e.utils import check_model_output_exists
 
-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/test_llama_pretrain.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/test_llama_vision.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/test_lora_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/test_mamba.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/test_mistral.py

@@ -8,7 +8,7 @@ import unittest
 
 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/test_mixtral.py

@@ -9,7 +9,7 @@ import unittest
 import torch
 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/test_optimizers.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/test_packing_loss.py

@@ -8,7 +8,7 @@ import unittest
 
 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/test_phi.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/test_relora_llama.py

@@ -7,7 +7,7 @@ import os
 import unittest
 from pathlib import Path
 
-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/test_reward_model_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