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 |
-
|
|
|
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 |
-
|
|
|
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 |
-
|
|
|
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 |
-
|
|
|
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 |
-
|
|
|
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
|