diff --git a/tests/e2e/test_fused_llama.py b/tests/e2e/test_fused_llama.py index 6979707c3..b4e23c6cc 100644 --- a/tests/e2e/test_fused_llama.py +++ b/tests/e2e/test_fused_llama.py @@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs from axolotl.train import train from axolotl.utils.config import normalize_config from axolotl.utils.dict import DictDefault -from tests.utils import with_temp_dir + +from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" diff --git a/tests/e2e/test_lora_llama.py b/tests/e2e/test_lora_llama.py index dcdfdb0ec..c74749535 100644 --- a/tests/e2e/test_lora_llama.py +++ b/tests/e2e/test_lora_llama.py @@ -12,7 +12,8 @@ from axolotl.common.cli import TrainerCliArgs from axolotl.train import train from axolotl.utils.config import normalize_config from axolotl.utils.dict import DictDefault -from tests.utils import with_temp_dir + +from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" diff --git a/tests/e2e/test_mistral.py b/tests/e2e/test_mistral.py index 7db40f005..c792b7d6e 100644 --- a/tests/e2e/test_mistral.py +++ b/tests/e2e/test_mistral.py @@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs from axolotl.train import train from axolotl.utils.config import normalize_config from axolotl.utils.dict import DictDefault -from tests.utils import with_temp_dir + +from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" diff --git a/tests/e2e/test_mistral_samplepack.py b/tests/e2e/test_mistral_samplepack.py index cf1aa57f5..248364fd9 100644 --- a/tests/e2e/test_mistral_samplepack.py +++ b/tests/e2e/test_mistral_samplepack.py @@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs from axolotl.train import train from axolotl.utils.config import normalize_config from axolotl.utils.dict import DictDefault -from tests.utils import with_temp_dir + +from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" diff --git a/tests/e2e/test_phi.py b/tests/e2e/test_phi.py index 4f121038d..6a29400e4 100644 --- a/tests/e2e/test_phi.py +++ b/tests/e2e/test_phi.py @@ -12,7 +12,8 @@ from axolotl.common.cli import TrainerCliArgs from axolotl.train import train from axolotl.utils.config import normalize_config from axolotl.utils.dict import DictDefault -from tests.utils import with_temp_dir + +from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" diff --git a/tests/utils.py b/tests/e2e/utils.py similarity index 100% rename from tests/utils.py rename to tests/e2e/utils.py