Enable saving run status by default in hyperparameter search pipelines

discovery-unicamp · Nov 11, 2024 · 53c1a76 · 53c1a76
1 parent 2e68956
commit 53c1a76
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/minerva/pipelines/hyperopt_hyperparameter_search.py b/minerva/pipelines/hyperopt_hyperparameter_search.py
@@ -24,7 +24,7 @@ def __init__(
         model: type,
         search_space: Dict[str, Any],
         log_dir: Optional[PathLike] = None,
-        save_run_status: bool = False,
+        save_run_status: bool = True,
     ):
         super().__init__(log_dir=log_dir, save_run_status=save_run_status)
         self.model = model

diff --git a/minerva/pipelines/ray_hyperparameter_search.py b/minerva/pipelines/ray_hyperparameter_search.py
@@ -21,7 +21,7 @@ def __init__(
         model: type,
         search_space: Dict[str, Any],
         log_dir: Optional[PathLike] = None,
-        save_run_status: bool = False,
+        save_run_status: bool = True,
     ):
         super().__init__(log_dir=log_dir, save_run_status=save_run_status)
         self.model = model