diff --git a/tests/e2e/integrations/test_cut_cross_entropy.py b/tests/e2e/integrations/test_cut_cross_entropy.py index 6562af176..c976ef5b7 100644 --- 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.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 diff --git a/tests/e2e/integrations/test_liger.py b/tests/e2e/integrations/test_liger.py index 9154bf9b8..6dc3108fb 100644 --- a/tests/e2e/integrations/test_liger.py +++ b/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 diff --git a/tests/e2e/patched/test_4d_multipack_llama.py b/tests/e2e/patched/test_4d_multipack_llama.py index 08b3bf0da..3cbe71450 100644 --- a/tests/e2e/patched/test_4d_multipack_llama.py +++ b/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 diff --git a/tests/e2e/patched/test_cli_integrations.py b/tests/e2e/patched/test_cli_integrations.py index 6ca7c52ae..ce9396d5f 100644 --- a/tests/e2e/patched/test_cli_integrations.py +++ b/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 diff --git a/tests/e2e/patched/test_fa_xentropy.py b/tests/e2e/patched/test_fa_xentropy.py index 791d955b2..70de4cd4b 100644 --- a/tests/e2e/patched/test_fa_xentropy.py +++ b/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 diff --git a/tests/e2e/patched/test_falcon_samplepack.py b/tests/e2e/patched/test_falcon_samplepack.py index 69516810f..b2cc420fa 100644 --- a/tests/e2e/patched/test_falcon_samplepack.py +++ b/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 diff --git a/tests/e2e/patched/test_fused_llama.py b/tests/e2e/patched/test_fused_llama.py index 23a0adfc0..7968274ba 100644 --- a/tests/e2e/patched/test_fused_llama.py +++ b/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 diff --git a/tests/e2e/patched/test_llama_s2_attention.py b/tests/e2e/patched/test_llama_s2_attention.py index d0fdd918a..50cd0b8a0 100644 --- a/tests/e2e/patched/test_llama_s2_attention.py +++ b/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 diff --git a/tests/e2e/patched/test_lora_llama_multipack.py b/tests/e2e/patched/test_lora_llama_multipack.py index 634e544d2..8cf6c2e57 100644 --- a/tests/e2e/patched/test_lora_llama_multipack.py +++ b/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 diff --git a/tests/e2e/patched/test_mistral_samplepack.py b/tests/e2e/patched/test_mistral_samplepack.py index e93863e09..1a8ce8662 100644 --- a/tests/e2e/patched/test_mistral_samplepack.py +++ b/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 diff --git a/tests/e2e/patched/test_mixtral_samplepack.py b/tests/e2e/patched/test_mixtral_samplepack.py index f87c34fd1..97400f598 100644 --- a/tests/e2e/patched/test_mixtral_samplepack.py +++ b/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 diff --git a/tests/e2e/patched/test_phi_multipack.py b/tests/e2e/patched/test_phi_multipack.py index 852ac7bec..7e8b549f9 100644 --- a/tests/e2e/patched/test_phi_multipack.py +++ b/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 diff --git a/tests/e2e/patched/test_resume.py b/tests/e2e/patched/test_resume.py index 5639d2eae..3f9551dd4 100644 --- a/tests/e2e/patched/test_resume.py +++ b/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 diff --git a/tests/e2e/patched/test_unsloth_qlora.py b/tests/e2e/patched/test_unsloth_qlora.py index 492bc1c23..a083a665f 100644 --- a/tests/e2e/patched/test_unsloth_qlora.py +++ b/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 diff --git a/tests/e2e/test_dpo.py b/tests/e2e/test_dpo.py index f8109373a..6a17e2c44 100644 --- a/tests/e2e/test_dpo.py +++ b/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 diff --git a/tests/e2e/test_embeddings_lr.py b/tests/e2e/test_embeddings_lr.py index 222d620ae..430dcfb51 100644 --- a/tests/e2e/test_embeddings_lr.py +++ b/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 diff --git a/tests/e2e/test_falcon.py b/tests/e2e/test_falcon.py index 117de6635..e82c1a709 100644 --- a/tests/e2e/test_falcon.py +++ b/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 diff --git a/tests/e2e/test_llama.py b/tests/e2e/test_llama.py index 4384bb61e..cb9eb6d44 100644 --- 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.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 diff --git a/tests/e2e/test_llama_pretrain.py b/tests/e2e/test_llama_pretrain.py index d13b10659..8faf28b96 100644 --- a/tests/e2e/test_llama_pretrain.py +++ b/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 diff --git a/tests/e2e/test_llama_vision.py b/tests/e2e/test_llama_vision.py index 250cf418c..5194f446a 100644 --- a/tests/e2e/test_llama_vision.py +++ b/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 diff --git a/tests/e2e/test_lora_llama.py b/tests/e2e/test_lora_llama.py index a7ead64a5..9b27ce8d0 100644 --- a/tests/e2e/test_lora_llama.py +++ b/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 diff --git a/tests/e2e/test_mamba.py b/tests/e2e/test_mamba.py index a1fc30862..a61d3790a 100644 --- a/tests/e2e/test_mamba.py +++ b/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 diff --git a/tests/e2e/test_mistral.py b/tests/e2e/test_mistral.py index 2e79fec8d..359a490f8 100644 --- a/tests/e2e/test_mistral.py +++ b/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 diff --git a/tests/e2e/test_mixtral.py b/tests/e2e/test_mixtral.py index 6792d05a6..103305925 100644 --- a/tests/e2e/test_mixtral.py +++ b/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 diff --git a/tests/e2e/test_optimizers.py b/tests/e2e/test_optimizers.py index f1bbaafd5..d0805bca3 100644 --- a/tests/e2e/test_optimizers.py +++ b/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 diff --git a/tests/e2e/test_packing_loss.py b/tests/e2e/test_packing_loss.py index dd0af32f3..748f818a7 100644 --- a/tests/e2e/test_packing_loss.py +++ b/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 diff --git a/tests/e2e/test_phi.py b/tests/e2e/test_phi.py index 7a08d0c6f..e5d0e86e7 100644 --- a/tests/e2e/test_phi.py +++ b/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 diff --git a/tests/e2e/test_relora_llama.py b/tests/e2e/test_relora_llama.py index fef6a3d30..2bb87c15b 100644 --- a/tests/e2e/test_relora_llama.py +++ b/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 diff --git a/tests/e2e/test_reward_model_llama.py b/tests/e2e/test_reward_model_llama.py index c4cb705ea..9617729bc 100644 --- a/tests/e2e/test_reward_model_llama.py +++ b/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