From 53c1a7628cd3e7b73fc3ef8d1f0bf16b0042646e Mon Sep 17 00:00:00 2001 From: Gabriel Gutierrez Date: Mon, 11 Nov 2024 09:52:58 -0300 Subject: [PATCH] Enable saving run status by default in hyperparameter search pipelines --- minerva/pipelines/hyperopt_hyperparameter_search.py | 2 +- minerva/pipelines/ray_hyperparameter_search.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/minerva/pipelines/hyperopt_hyperparameter_search.py b/minerva/pipelines/hyperopt_hyperparameter_search.py index 3d7f5fb..0cafb0c 100644 --- 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 index 095d779..c8b96c2 100644 --- 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