From a3f156c45245e12f14a66d7233312ffb7e9de6dc Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 25 Jun 2025 19:36:01 +0200 Subject: [PATCH] refactor(qav3): refine typing MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../freqaimodels/QuickAdapterRegressorV3.py | 12 ++++++------ quickadapter/user_data/strategies/QuickAdapterV3.py | 6 +++--- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index ec83444..dc907f7 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -1144,9 +1144,9 @@ def train_objective( X_test = X_test.iloc[-test_window:] y_test = y_test.iloc[-test_window:] test_extrema = y_test.get(EXTREMA_COLUMN) - n_test_minima = sp.signal.find_peaks(-test_extrema)[0].size - n_test_maxima = sp.signal.find_peaks(test_extrema)[0].size - n_test_extrema = n_test_minima + n_test_maxima + n_test_minima: int = sp.signal.find_peaks(-test_extrema)[0].size + n_test_maxima: int = sp.signal.find_peaks(test_extrema)[0].size + n_test_extrema: int = n_test_minima + n_test_maxima min_test_extrema: int = calculate_min_extrema( test_window, fit_live_predictions_candles ) @@ -1182,9 +1182,9 @@ def train_objective( X = X.iloc[-train_window:] y = y.iloc[-train_window:] train_extrema = y.get(EXTREMA_COLUMN) - n_train_minima = sp.signal.find_peaks(-train_extrema)[0].size - n_train_maxima = sp.signal.find_peaks(train_extrema)[0].size - n_train_extrema = n_train_minima + n_train_maxima + n_train_minima: int = sp.signal.find_peaks(-train_extrema)[0].size + n_train_maxima: int = sp.signal.find_peaks(train_extrema)[0].size + n_train_extrema: int = n_train_minima + n_train_maxima min_train_extrema: int = calculate_min_extrema( train_window, fit_live_predictions_candles ) diff --git a/quickadapter/user_data/strategies/QuickAdapterV3.py b/quickadapter/user_data/strategies/QuickAdapterV3.py index 07a6bd5..72e4695 100644 --- a/quickadapter/user_data/strategies/QuickAdapterV3.py +++ b/quickadapter/user_data/strategies/QuickAdapterV3.py @@ -456,9 +456,9 @@ class QuickAdapterV3(IStrategy): ) if debug: logger.info(f"{dataframe[EXTREMA_COLUMN].to_numpy()=}") - n_minima = sp.signal.find_peaks(-dataframe[EXTREMA_COLUMN])[0].size - n_maxima = sp.signal.find_peaks(dataframe[EXTREMA_COLUMN])[0].size - n_extrema = n_minima + n_maxima + n_minima: int = sp.signal.find_peaks(-dataframe[EXTREMA_COLUMN])[0].size + n_maxima: int = sp.signal.find_peaks(dataframe[EXTREMA_COLUMN])[0].size + n_extrema: int = n_minima + n_maxima logger.info(f"{n_extrema=}") return dataframe -- 2.43.0