winglian commited on
Commit
c74f045
β€’
1 Parent(s): 0402d19

chore: lint

Browse files
tests/e2e/test_fused_llama.py CHANGED
@@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
- from tests.utils import with_temp_dir
 
18
 
19
  LOG = logging.getLogger("axolotl.tests.e2e")
20
  os.environ["WANDB_DISABLED"] = "true"
 
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
+
18
+ from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level
19
 
20
  LOG = logging.getLogger("axolotl.tests.e2e")
21
  os.environ["WANDB_DISABLED"] = "true"
tests/e2e/test_lora_llama.py CHANGED
@@ -12,7 +12,8 @@ from axolotl.common.cli import TrainerCliArgs
12
  from axolotl.train import train
13
  from axolotl.utils.config import normalize_config
14
  from axolotl.utils.dict import DictDefault
15
- from tests.utils import with_temp_dir
 
16
 
17
  LOG = logging.getLogger("axolotl.tests.e2e")
18
  os.environ["WANDB_DISABLED"] = "true"
 
12
  from axolotl.train import train
13
  from axolotl.utils.config import normalize_config
14
  from axolotl.utils.dict import DictDefault
15
+
16
+ from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level
17
 
18
  LOG = logging.getLogger("axolotl.tests.e2e")
19
  os.environ["WANDB_DISABLED"] = "true"
tests/e2e/test_mistral.py CHANGED
@@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
- from tests.utils import with_temp_dir
 
18
 
19
  LOG = logging.getLogger("axolotl.tests.e2e")
20
  os.environ["WANDB_DISABLED"] = "true"
 
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
+
18
+ from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level
19
 
20
  LOG = logging.getLogger("axolotl.tests.e2e")
21
  os.environ["WANDB_DISABLED"] = "true"
tests/e2e/test_mistral_samplepack.py CHANGED
@@ -14,7 +14,8 @@ from axolotl.common.cli import TrainerCliArgs
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
- from tests.utils import with_temp_dir
 
18
 
19
  LOG = logging.getLogger("axolotl.tests.e2e")
20
  os.environ["WANDB_DISABLED"] = "true"
 
14
  from axolotl.train import train
15
  from axolotl.utils.config import normalize_config
16
  from axolotl.utils.dict import DictDefault
17
+
18
+ from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level
19
 
20
  LOG = logging.getLogger("axolotl.tests.e2e")
21
  os.environ["WANDB_DISABLED"] = "true"
tests/e2e/test_phi.py CHANGED
@@ -12,7 +12,8 @@ from axolotl.common.cli import TrainerCliArgs
12
  from axolotl.train import train
13
  from axolotl.utils.config import normalize_config
14
  from axolotl.utils.dict import DictDefault
15
- from tests.utils import with_temp_dir
 
16
 
17
  LOG = logging.getLogger("axolotl.tests.e2e")
18
  os.environ["WANDB_DISABLED"] = "true"
 
12
  from axolotl.train import train
13
  from axolotl.utils.config import normalize_config
14
  from axolotl.utils.dict import DictDefault
15
+
16
+ from .utils import with_temp_dir # pylint: disable=relative-beyond-top-level
17
 
18
  LOG = logging.getLogger("axolotl.tests.e2e")
19
  os.environ["WANDB_DISABLED"] = "true"
tests/{utils.py β†’ e2e/utils.py} RENAMED
File without changes