tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

remove deprecated wandb env var (#2751)

Browse Source 
* remove deprecated wandb env var

* remove os.environ wandb setting; unused loggers

* remove os.environ wandb setting; unused loggers
This commit is contained in:
Dan Saunders
2025-06-03 16:04:15 -05:00
committed by GitHub
parent 2bf61d8e25
commit 1d91d905c9
46 changed files with 3 additions and 232 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/e2e/patched/test_4d_multipack_llama.py
View File

@@ -2,7 +2,6 @@
E2E tests for multipack fft llama using 4d attention masks
"""
import os
import unittest
from axolotl.cli.args import TrainerCliArgs
@@ -10,13 +9,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class Test4dMultipackLlama(unittest.TestCase):
"""

6
tests/e2e/patched/test_fa_xentropy.py
View File

@@ -2,8 +2,6 @@
E2E tests for lora llama
"""
import os
import pytest
from transformers.utils import is_torch_bf16_gpu_available
@@ -12,13 +10,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, check_tensorboard
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestFAXentropyLlama:
"""

5
tests/e2e/patched/test_falcon_samplepack.py
View File

@@ -2,7 +2,6 @@
E2E tests for falcon
"""
import os
import unittest
import pytest
@@ -12,13 +11,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestFalconPatched(unittest.TestCase):
"""

5
tests/e2e/patched/test_fused_llama.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
import pytest
@@ -13,13 +12,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
@pytest.mark.skip("FIXME, mostly underused functionality")
class TestFusedLlama(unittest.TestCase):

5
tests/e2e/patched/test_llama_s2_attention.py
View File

@@ -2,7 +2,6 @@
E2E tests for llama w/ S2 attn
"""
import os
import unittest
import pytest
@@ -12,13 +11,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
@pytest.mark.skip(reason="FIXME?")
class TestLlamaShiftedSparseAttention(unittest.TestCase):

5
tests/e2e/patched/test_lora_llama_multipack.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
import pytest
@@ -13,13 +12,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestLoraLlama(unittest.TestCase):
"""

5
tests/e2e/patched/test_mistral_samplepack.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
from axolotl.cli.args import TrainerCliArgs
@@ -10,13 +9,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestMistral(unittest.TestCase):
"""

5
tests/e2e/patched/test_mixtral_samplepack.py
View File

@@ -2,7 +2,6 @@
E2E tests for mixtral
"""
import os
import unittest
from axolotl.cli.args import TrainerCliArgs
@@ -10,13 +9,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestMixtral(unittest.TestCase):
"""

5
tests/e2e/patched/test_phi_multipack.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
from axolotl.cli.args import TrainerCliArgs
@@ -10,13 +9,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestPhiMultipack(unittest.TestCase):
"""

5
tests/e2e/patched/test_resume.py
View File

@@ -2,7 +2,6 @@
E2E tests for resuming training
"""
import os
import re
import subprocess
@@ -13,13 +12,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, most_recent_subdir, require_torch_2_6_0
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestResumeLlama:
"""

6
tests/e2e/patched/test_unsloth_qlora.py
View File

@@ -2,8 +2,6 @@
e2e tests for unsloth qlora
"""
import os
import pytest
from axolotl.cli.args import TrainerCliArgs
@@ -11,13 +9,9 @@ from axolotl.common.datasets import load_datasets
from axolotl.train import train
from axolotl.utils.config import normalize_config, validate_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_model_output_exists, check_tensorboard
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
# pylint: disable=duplicate-code
@pytest.mark.skip(