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

3
tests/e2e/multigpu/patched/test_sp.py
View File

@@ -1,6 +1,5 @@
"""E2E tests for sequence parallelism"""
import os
from pathlib import Path
import pytest
@@ -12,8 +11,6 @@ from axolotl.utils.dict import DictDefault
from ...utils import check_tensorboard
os.environ["WANDB_DISABLED"] = "true"
class TestSequenceParallelism:
"""Test case for training with sequence parallelism enabled"""

5
tests/e2e/multigpu/solo/test_flex.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu lora tinyllama
"""
import os
from pathlib import Path
import pytest
@@ -13,13 +12,9 @@ from transformers.testing_utils import get_torch_dist_unique_port
from transformers.utils import is_torch_bf16_gpu_available
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from tests.e2e.utils import check_tensorboard, require_torch_2_6_0
LOG = get_logger("axolotl.tests.e2e.multigpu")
os.environ["WANDB_DISABLED"] = "true"
AXOLOTL_ROOT = Path(__file__).parent.parent.parent.parent

5
tests/e2e/multigpu/test_eval.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu eval
"""
import os
from pathlib import Path
import yaml
@@ -10,13 +9,9 @@ from accelerate.test_utils import execute_subprocess_async
from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from ..utils import check_tensorboard
LOG = get_logger("axolotl.tests.e2e.multigpu")
os.environ["WANDB_DISABLED"] = "true"
AXOLOTL_ROOT = Path(__file__).parent.parent.parent.parent

5
tests/e2e/multigpu/test_gemma3.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu lora tinyllama
"""
import os
from pathlib import Path
import pytest
@@ -12,13 +11,9 @@ from huggingface_hub import snapshot_download
from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from tests.e2e.utils import check_tensorboard
LOG = get_logger("axolotl.tests.e2e.multigpu")
os.environ["WANDB_DISABLED"] = "true"
AXOLOTL_ROOT = Path(__file__).parent.parent.parent.parent

5
tests/e2e/multigpu/test_llama.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu lora tinyllama
"""
import os
from pathlib import Path
import pytest
@@ -14,13 +13,9 @@ from packaging import version
from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from tests.e2e.utils import check_tensorboard, require_torch_2_6_0
LOG = get_logger("axolotl.tests.e2e.multigpu")
os.environ["WANDB_DISABLED"] = "true"
AXOLOTL_ROOT = Path(__file__).parent.parent.parent.parent

5
tests/e2e/multigpu/test_qwen2.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu qwen2
"""
import os
from pathlib import Path
import pytest
@@ -11,10 +10,6 @@ from accelerate.test_utils import execute_subprocess_async
from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
LOG = get_logger("axolotl.tests.e2e.multigpu")
os.environ["WANDB_DISABLED"] = "true"
class TestMultiGPUQwen2:

5
tests/e2e/multigpu/test_ray.py
View File

@@ -2,7 +2,6 @@
E2E tests for multigpu post-training use Ray Train
"""
import os
from pathlib import Path
import pytest
@@ -10,13 +9,9 @@ import yaml
from accelerate.test_utils import execute_subprocess_async
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
from tests.e2e.utils import check_tensorboard, require_torch_lt_2_6_0
LOG = get_logger(__name__)
os.environ["WANDB_DISABLED"] = "true"
AXOLOTL_ROOT = Path(__file__).parent.parent.parent.parent

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(

5
tests/e2e/solo/test_flex.py
View File

@@ -2,7 +2,6 @@
E2E tests for packed training w/ flex attention
"""
import os
import unittest
from transformers.utils import is_torch_bf16_gpu_available
@@ -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_tensorboard, require_torch_2_6_0, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestPackedFlex(unittest.TestCase):
"""

5
tests/e2e/solo/test_relora_llama.py
View File

@@ -2,7 +2,6 @@
E2E tests for relora llama
"""
import os
import unittest
from pathlib import Path
@@ -11,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, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestReLoraLlama(unittest.TestCase):
"""

5
tests/e2e/test_deepseekv3.py
View File

@@ -2,7 +2,6 @@
E2E tests for deepseekv3
"""
import os
from pathlib import Path
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 tests.hf_offline_utils import enable_hf_offline
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestDeepseekV3:
"""

5
tests/e2e/test_dpo.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
from pathlib import Path
@@ -13,13 +12,9 @@ from axolotl.common.datasets import load_preference_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 TestDPOLlamaLora(unittest.TestCase):
"""

5
tests/e2e/test_embeddings_lr.py
View File

@@ -2,7 +2,6 @@
E2E tests for llama pretrain
"""
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, check_tensorboard, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestEmbeddingsLrScale(unittest.TestCase):
"""

3
tests/e2e/test_evaluate.py
View File

@@ -1,6 +1,5 @@
"""E2E smoke test for evaluate CLI command"""
import os
from pathlib import Path
import yaml
@@ -9,8 +8,6 @@ from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
os.environ["WANDB_DISABLED"] = "true"
class TestE2eEvaluate:
"""Test cases for evaluate CLI"""

5
tests/e2e/test_falcon.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 TestFalcon(unittest.TestCase):
"""

5
tests/e2e/test_gemma2.py
View File

@@ -2,7 +2,6 @@
E2E tests for gemma2
"""
import os
from pathlib import Path
import pytest
@@ -12,10 +11,6 @@ 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
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestGemma2:

5
tests/e2e/test_gemma3_text.py
View File

@@ -2,7 +2,6 @@
E2E tests for gemma3_text
"""
import os
from pathlib import Path
import pytest
@@ -12,10 +11,6 @@ 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
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestGemma3Text:

6
tests/e2e/test_llama.py
View File

@@ -2,20 +2,14 @@
E2E tests for llama
"""
import os
from axolotl.cli.args import TrainerCliArgs
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 tests.e2e.utils import check_model_output_exists
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestLlama:
"""

6
tests/e2e/test_llama_pretrain.py
View File

@@ -2,8 +2,6 @@
E2E tests for llama pretrain
"""
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"
class TestPretrainLlama:
"""

5
tests/e2e/test_llama_vision.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 TestLlamaVision(unittest.TestCase):
"""

5
tests/e2e/test_lora_llama.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 TestLoraLlama(unittest.TestCase):
"""

5
tests/e2e/test_mamba.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
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="skipping until upstreamed into transformers")
class TestMamba(unittest.TestCase):

5
tests/e2e/test_mistral.py
View File

@@ -2,7 +2,6 @@
E2E tests for lora llama
"""
import os
import unittest
from transformers.utils import is_torch_bf16_gpu_available
@@ -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 TestMistral(unittest.TestCase):
"""

5
tests/e2e/test_mixtral.py
View File

@@ -2,7 +2,6 @@
E2E tests for mixtral
"""
import os
import unittest
import torch
@@ -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 TestMixtral(unittest.TestCase):
"""

5
tests/e2e/test_optimizers.py
View File

@@ -2,7 +2,6 @@
E2E tests for custom optimizers using 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, require_torch_2_5_1, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestCustomOptimizers(unittest.TestCase):
"""

5
tests/e2e/test_packing_loss.py
View File

@@ -2,7 +2,6 @@
E2E tests for packed training
"""
import os
import unittest
from transformers.utils import is_torch_bf16_gpu_available
@@ -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_tensorboard, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestPackedLlama(unittest.TestCase):
"""

5
tests/e2e/test_phi.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 TestPhi(unittest.TestCase):
"""

5
tests/e2e/test_process_reward_model_smollm2.py
View File

@@ -2,7 +2,6 @@
E2E tests for process reward model w/ 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, check_tensorboard, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestProcessRewardSmolLM2(unittest.TestCase):
"""

5
tests/e2e/test_qwen.py
View File

@@ -2,7 +2,6 @@
E2E tests for qwen
"""
import os
from pathlib import Path
import pytest
@@ -11,10 +10,6 @@ from accelerate.test_utils import execute_subprocess_async
from transformers.testing_utils import get_torch_dist_unique_port
from axolotl.utils.dict import DictDefault
from axolotl.utils.logging import get_logger
LOG = get_logger("axolotl.tests.qwen")
os.environ["WANDB_DISABLED"] = "true"
class TestE2eQwen:

5
tests/e2e/test_reward_model_smollm2.py
View File

@@ -2,7 +2,6 @@
E2E tests for reward model 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, check_tensorboard, with_temp_dir
LOG = get_logger("axolotl.tests.e2e")
os.environ["WANDB_DISABLED"] = "true"
class TestRewardModelLoraSmolLM2(unittest.TestCase):
"""

5
tests/e2e/test_schedulers.py
View File

@@ -2,7 +2,6 @@
E2E tests for custom schedulers using 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 TestCustomSchedulers(unittest.TestCase):
"""