From c74f045ba7b755910b57229d3144e2f0a7ce769f Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sun, 5 Nov 2023 19:49:31 -0500
Subject: [PATCH] chore: lint

---
 tests/e2e/test_fused_llama.py        | 3 ++-
 tests/e2e/test_lora_llama.py         | 3 ++-
 tests/e2e/test_mistral.py            | 3 ++-
 tests/e2e/test_mistral_samplepack.py | 3 ++-
 tests/e2e/test_phi.py                | 3 ++-
 tests/{ => e2e}/utils.py             | 0
 6 files changed, 10 insertions(+), 5 deletions(-)
 rename tests/{ => e2e}/utils.py (100%)

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