fixes

vturrisi · Sep 27, 2021 · 13234df · 13234df
1 parent 2c869f0
commit 13234df
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/args/test_setup.py b/tests/args/test_setup.py
@@ -98,6 +98,9 @@ def test_setup_pretrain():
         "--final_tau_momentum",
         "1.0",
         "--momentum_classifier",
+        "--wandb",
+        "--save_checkpoint",
+        "--auto_umap",
     ]
     # Write string to a file
     with open("dummy_script.py", "w") as f: