refactor(qav3): rename storage optuna tunable

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

diff --git a/quickadapter/user_data/config-template.json b/quickadapter/user_data/config-template.json
index ba7ba0e931fbee7dbe9052724067283cc8e9528a..c1d4c4f5d45546d9e403bd61141431a3e2fccea1 100644 (file)
--- a/quickadapter/user_data/config-template.json
+++ b/quickadapter/user_data/config-template.json
@@ -122,7 +122,7 @@
       "n_jobs": 6,
       "timeout": 3600,
       "candles_step": 10,
-      "storage_type": "sqlite"
+      "storage": "sqlite"
     },
     "extra_returns_per_train": {
       "DI_value_param1": 0,

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index 86318b62d542ad36f95f0c08ffd72de9b7cb5c4d..cc0ee08bcae2b26765ce57c46347f88403281574 100644 (file)
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -234,10 +234,10 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
 
     def get_optuna_storage(self, dk: FreqaiDataKitchen):
         storage_dir = str(dk.full_path)
-        storage_type = self.__optuna_config.get("storage_type", "sqlite")
-        if storage_type == "sqlite":
+        storage_backend = self.__optuna_config.get("storage", "sqlite")
+        if storage_backend == "sqlite":
             storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite"
-        elif storage_type == "file":
+        elif storage_backend == "file":
             storage = optuna.storages.JournalStorage(
                 optuna.storages.journal.JournalFileBackend(
                     f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log"

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index 36363c87d8230114953683df19781ea6335bc412..63d89b8035e54c2458874216c5b837545cbac05c 100644 (file)
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -237,10 +237,10 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
 
     def get_optuna_storage(self, dk: FreqaiDataKitchen):
         storage_dir = str(dk.full_path)
-        storage_type = self.__optuna_config.get("storage_type", "sqlite")
-        if storage_type == "sqlite":
+        storage_backend = self.__optuna_config.get("storage", "sqlite")
+        if storage_backend == "sqlite":
             storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite"
-        elif storage_type == "file":
+        elif storage_backend == "file":
             storage = optuna.storages.JournalStorage(
                 optuna.storages.journal.JournalFileBackend(
                     f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log"