From 7eab1caa1df29fa84a473628d73ce16d978d0c13 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 5 Feb 2025 22:17:55 +0100 Subject: [PATCH] refactor(qav3): add typing MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../freqaimodels/LightGBMRegressorQuickAdapterV35.py | 4 +++- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index d7cdc46..ff5dbcf 100644 --- 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 3f8f66c..33188d8 100644 --- 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(): -- 2.43.0