remove deprecated wandb env var (#2751)

* remove deprecated wandb env var

* remove os.environ wandb setting; unused loggers

* remove os.environ wandb setting; unused loggers

tests/e2e/multigpu/patched/test_sp.py

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

tests/e2e/multigpu/solo/test_flex.py

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

tests/e2e/multigpu/test_eval.py

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

tests/e2e/multigpu/test_gemma3.py

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

tests/e2e/multigpu/test_llama.py

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

tests/e2e/multigpu/test_qwen2.py

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

tests/e2e/multigpu/test_ray.py

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