adding an arg for data_path

deepskies · May 7, 2024 · 0284186 · 0284186
1 parent d0c4359
commit 0284186
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_DeepEvidentialRegression.py b/test/test_DeepEvidentialRegression.py
@@ -117,8 +117,12 @@ def test_DER_chkpt_saved(self,
         subprocess_args = [
             "python",
             "src/scripts/DeepEvidentialRegression.py",
+            "--data_path",
+            str(temp_data),
             "--noise_level",
             noise_level,
+            "--size_df",
+            size_df,
             "--out_dir",
             str(temp_directory) + "/",
             "--n_epochs",