From bfc838de0ab39ddc305b34b9c9573cd0ad73bf1c Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Fri, 11 Apr 2025 17:46:09 +0200 Subject: [PATCH] refactor(qav3): align optuna results helper namespace 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 | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 414610d..cf48014 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -143,12 +143,12 @@ class QuickAdapterRegressorV3(BaseRegressionModel): def get_optuna_value(self, pair: str, namespace: str) -> float: if namespace == "hp": - rmse = self._optuna_hp_value.get(pair) + value = self._optuna_hp_value.get(pair) elif namespace == "train": - rmse = self._optuna_train_value.get(pair) + value = self._optuna_train_value.get(pair) else: raise ValueError(f"Invalid namespace: {namespace}") - return rmse + return value def set_optuna_value(self, pair: str, namespace: str, value: float) -> None: if namespace == "hp": -- 2.43.0