refactor(qav3): add typing

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

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index d7cdc4622ffa2d72ec03c38f74843d4b1273e3e6..ff5dbcf4c3146080a375b95cccdb7f7138c018fd 100644 (file)
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -199,7 +199,9 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
         return eval_set, eval_weights
 
 
-def min_max_pred(pred_df, fit_live_predictions_candles, label_period_candles):
+def min_max_pred(
+    pred_df: pd.Dataframe, fit_live_predictions_candles: int, label_period_candles: int
+):
     pred_df_sorted = pd.DataFrame()
     for label in pred_df.keys():
         if pred_df[label].dtype == object:

diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
index 3f8f66c68621d8be32d4fae1df4a4e88595d9749..33188d81e5dc1a35b5e4adbd4362739472d9e1d5 100644 (file)
--- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py
@@ -200,7 +200,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
 
 
 def min_max_pred(
-    pred_df: pd.DataFrame, fit_live_predictions_candles, label_period_candles
+    pred_df: pd.DataFrame, fit_live_predictions_candles: int, label_period_candles: int
 ):
     pred_df_sorted = pd.DataFrame()
     for label in pred_df.keys():