test(tests/test_train.py): disable wandb during testing

AdityaNG · May 8, 2024 · c58d430 · c58d430
1 parent 4d3f64f
commit c58d430
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_train.py b/tests/test_train.py
@@ -51,5 +51,7 @@ class Args:
         architecture = "MLP"
         device = "cpu"
 
+    os.environ["WANDB_DISABLED"] = "true"
+
     args = Args()
     main(args)