From 633952f9d2d32bb837cbe210336047f5b10ad228 Mon Sep 17 00:00:00 2001 From: golmschenk Date: Thu, 10 Oct 2024 14:07:51 -0400 Subject: [PATCH] Use both old and new wandb disabling methods to cover multiple versions during tests --- tests/end_to_end_tests/test_toy_train_session.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/end_to_end_tests/test_toy_train_session.py b/tests/end_to_end_tests/test_toy_train_session.py index 47e17f0d..17c521db 100644 --- a/tests/end_to_end_tests/test_toy_train_session.py +++ b/tests/end_to_end_tests/test_toy_train_session.py @@ -12,6 +12,7 @@ def test_toy_train_session(): os.environ["WANDB_MODE"] = "disabled" + os.environ['WANDB_DISABLED'] = 'true' model = SingleDenseLayerBinaryClassificationModel.new(input_size=100) dataset = get_toy_dataset() dataset.post_injection_transform = partial(