From 3b7b47e5a8a2d700d00a9cfe7a4edad7aae55f61 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 16 Jun 2025 14:06:42 +0200 Subject: [PATCH] perf(qav3): speed up pivots confirmation use case MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py | 2 ++ quickadapter/user_data/strategies/Utils.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 6f0c72c..47f569f 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -1344,6 +1344,8 @@ def zigzag( def calculate_min_slopes_ok(pos: int, slopes_ok_threshold: float) -> int: min_slopes_ok, max_slopes_ok = calculate_slopes_ok_min_max(slopes_ok_threshold) + if min_slopes_ok == max_slopes_ok: + return min_slopes_ok volatility_quantile = calculate_volatility_quantile(pos) if np.isnan(volatility_quantile): return int(round(median([min_slopes_ok, max_slopes_ok]))) diff --git a/quickadapter/user_data/strategies/Utils.py b/quickadapter/user_data/strategies/Utils.py index 0d91aea..f598b79 100644 --- a/quickadapter/user_data/strategies/Utils.py +++ b/quickadapter/user_data/strategies/Utils.py @@ -473,6 +473,8 @@ def zigzag( def calculate_min_slopes_ok(pos: int, slopes_ok_threshold: float) -> int: min_slopes_ok, max_slopes_ok = calculate_slopes_ok_min_max(slopes_ok_threshold) + if min_slopes_ok == max_slopes_ok: + return min_slopes_ok volatility_quantile = calculate_volatility_quantile(pos) if np.isnan(volatility_quantile): return int(round(median([min_slopes_ok, max_slopes_ok]))) -- 2.43.0