refactor(qav3): factor out optuna storage handling ops

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 34a3acef1ad50e5bff6fd91f6f5779d5f1e85629..dee1983630246752d0cf5f12c7276ba019e95152 100644 (file)
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -305,7 +305,8 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
         study_name = f"hp-{dk.pair}"
         storage = self.optuna_storage(dk)
         pruner = optuna.pruners.HyperbandPruner()
-        previous_study = self.optuna_study_load_and_cleanup(study_name, storage)
+        previous_study = self.optuna_study_load(study_name, storage)
+        self.optuna_study_delete(study_name, storage)
         study = optuna.create_study(
             study_name=study_name,
             sampler=optuna.samplers.TPESampler(
@@ -375,7 +376,8 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
         study_name = f"period-{dk.pair}"
         storage = self.optuna_storage(dk)
         pruner = optuna.pruners.HyperbandPruner()
-        previous_study = self.optuna_study_load_and_cleanup(study_name, storage)
+        previous_study = self.optuna_study_load(study_name, storage)
+        self.optuna_study_delete(study_name, storage)
         study = optuna.create_study(
             study_name=study_name,
             sampler=optuna.samplers.TPESampler(
@@ -443,17 +445,17 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
                 return json.load(read_file)
         return None
 
-    def optuna_study_load_and_cleanup(
-        self, study_name: str, storage
-    ) -> optuna.study.Study | None:
-        try:
-            study = optuna.load_study(study_name=study_name, storage=storage)
-        except Exception:
-            study = None
+    def optuna_study_delete(self, study_name: str, storage) -> None:
         try:
             optuna.delete_study(study_name=study_name, storage=storage)
         except Exception:
             pass
+
+    def optuna_study_load(self, study_name: str, storage) -> optuna.study.Study | None:
+        try:
+            study = optuna.load_study(study_name=study_name, storage=storage)
+        except Exception:
+            study = None
         return study
 
     def optuna_study_has_best_params(self, study: optuna.study.Study | None) -> bool:

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index b22cbe17eaf0b27f261a241a7abc1702a17aaeda..2568a5b10f08443de7b238696f6213c957789e45 100644 (file)
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -306,7 +306,8 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
         study_name = f"hp-{dk.pair}"
         storage = self.optuna_storage(dk)
         pruner = optuna.pruners.HyperbandPruner()
-        previous_study = self.optuna_study_load_and_cleanup(study_name, storage)
+        previous_study = self.optuna_study_load(study_name, storage)
+        self.optuna_study_delete(study_name, storage)
         study = optuna.create_study(
             study_name=study_name,
             sampler=optuna.samplers.TPESampler(
@@ -376,7 +377,8 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
         study_name = f"period-{dk.pair}"
         storage = self.optuna_storage(dk)
         pruner = optuna.pruners.HyperbandPruner()
-        previous_study = self.optuna_study_load_and_cleanup(study_name, storage)
+        previous_study = self.optuna_study_load(study_name, storage)
+        self.optuna_study_delete(study_name, storage)
         study = optuna.create_study(
             study_name=study_name,
             sampler=optuna.samplers.TPESampler(
@@ -444,17 +446,17 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
                 return json.load(read_file)
         return None
 
-    def optuna_study_load_and_cleanup(
-        self, study_name: str, storage
-    ) -> optuna.study.Study | None:
-        try:
-            study = optuna.load_study(study_name=study_name, storage=storage)
-        except Exception:
-            study = None
+    def optuna_study_delete(self, study_name: str, storage) -> None:
         try:
             optuna.delete_study(study_name=study_name, storage=storage)
         except Exception:
             pass
+
+    def optuna_study_load(self, study_name: str, storage) -> optuna.study.Study | None:
+        try:
+            study = optuna.load_study(study_name=study_name, storage=storage)
+        except Exception:
+            study = None
         return study
 
     def optuna_study_has_best_params(self, study: optuna.study.Study | None) -> bool: