fix(qav3): ensure previous optuna study is loaded when continuous = False

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index 1081505c0ad51667b48a6e7d23ee1293f7192d50..4b83eb622f01d5fa6713be52b322c2638e410c5e 100644 (file)
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -318,6 +318,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
             pruner=pruner,
             direction=optuna.study.StudyDirection.MINIMIZE,
             storage=storage,
+            load_if_exists=not self.__optuna_config.get("continuous", True),
         )
         if self.__optuna_config.get("warm_start", True):
             self.optuna_hp_enqueue_previous_best_trial(pair, study)
@@ -396,6 +397,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
             pruner=pruner,
             direction=optuna.study.StudyDirection.MINIMIZE,
             storage=storage,
+            load_if_exists=not self.__optuna_config.get("continuous", True),
         )
         if self.__optuna_config.get("warm_start", True):
             self.optuna_period_enqueue_previous_best_trial(pair, study)

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index e7a46f5ed5980c3d681086c39c35716c8f8b10c8..79632455d389cbff8093ca996309d0226a62f237 100644 (file)
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -321,6 +321,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
             pruner=pruner,
             direction=optuna.study.StudyDirection.MINIMIZE,
             storage=storage,
+            load_if_exists=not self.__optuna_config.get("continuous", True),
         )
         if self.__optuna_config.get("warm_start", True):
             self.optuna_hp_enqueue_previous_best_trial(pair, study)
@@ -399,6 +400,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
             pruner=pruner,
             direction=optuna.study.StudyDirection.MINIMIZE,
             storage=storage,
+            load_if_exists=not self.__optuna_config.get("continuous", True),
         )
         if self.__optuna_config.get("warm_start", True):
             self.optuna_period_enqueue_previous_best_trial(pair, study)