all tests passing locally

deepskies · Apr 1, 2024 · 48480d3 · 48480d3
1 parent 5b9507c
commit 48480d3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_DeepEnsemble.py b/test/test_DeepEnsemble.py
@@ -104,9 +104,9 @@ def test_chkpt_saved(temp_directory):
         wd,
         "--n_epochs",
         str(n_epochs),
-        "--save_final_checkpoints",
-        "True",
-    ]
+        "--save_final_checkpoint",
+        "--savefig"
+        ]
     # now run the subprocess
     subprocess.run(subprocess_args, check=True)
     # check if the right number of checkpoints are saved