From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Wed, 12 Feb 2025 22:23:38 +0000 (+0100)
Subject: refactor(qav3): rename storage optuna tunable
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=a3b633f212bd947e5792404255c4d4c1e7ceedb8;p=freqai-strategies.git

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 ba7ba0e..c1d4c4f 100644
--- 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 86318b6..cc0ee08 100644
--- 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 36363c8..63d89b8 100644
--- 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"