From bded196f6c8827bbb5e2530c03cc62648b8b8369 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 2 Apr 2025 21:13:44 +0200 Subject: [PATCH] refactor(qav3): cleanups MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/QuickAdapterRegressorV3.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 8b0087b..1a991ef 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -320,12 +320,12 @@ class QuickAdapterRegressorV3(BaseRegressionModel): rmse_storage: dict[str, float], ) -> None: identifier = self.freqai_info.get("identifier") - study = self.optuna_create_study(f"{identifier}-{namespace}-{pair}", pair) + study = self.optuna_create_study(pair, f"{identifier}-{namespace}-{pair}") if not study: return if self.__optuna_config.get("warm_start"): - self.optuna_enqueue_previous_best_params(pair, study, namespace) + self.optuna_enqueue_previous_best_params(pair, namespace, study) logger.info(f"Optuna {namespace} hyperopt started") start_time = time.time() @@ -380,7 +380,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel): return storage def optuna_create_study( - self, study_name: str, pair: str + self, pair: str, study_name: str ) -> Optional[optuna.study.Study]: try: storage = self.optuna_storage(pair) @@ -412,7 +412,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel): return None def optuna_enqueue_previous_best_params( - self, pair: str, study: optuna.study.Study, namespace: str + self, pair: str, namespace: str, study: optuna.study.Study ) -> None: if namespace == "hp": best_params = self.__optuna_hp_params.get(pair, {}) -- 2.43.0