From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Wed, 2 Apr 2025 19:31:36 +0000 (+0200)
Subject: refactor(qav3): refine typing
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=34ab139340d654201cc1c84c16e1f4fdba42e804;p=freqai-strategies.git

refactor(qav3): refine typing

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

diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
index 1a991ef..f5948e7 100644
--- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
+++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
@@ -548,7 +548,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel):
         return min_pred[EXTREMA_COLUMN], max_pred[EXTREMA_COLUMN]
 
 
-def get_callbacks(trial: optuna.Trial, regressor: str) -> list:
+def get_callbacks(trial: optuna.Trial, regressor: str) -> list[Callable]:
     if regressor == "xgboost":
         callbacks = [
             optuna.integration.XGBoostPruningCallback(trial, "validation_0-rmse")
@@ -569,7 +569,7 @@ def train_regressor(
     eval_weights: Optional[list[np.ndarray]],
     model_training_parameters: dict,
     init_model: Any = None,
-    callbacks: list = None,
+    callbacks: list[Callable] = None,
 ) -> Any:
     if regressor == "xgboost":
         from xgboost import XGBRegressor