refactor(qav3): zigzag() trivial code cleanup

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 e2546bb04744d5de7886c0e106fc0905d60ffa49..9291f69eec9ca22e5c78a77e1138d367f40f6560 100644 (file)
--- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
+++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
@@ -961,7 +961,7 @@ def zigzag(
     else:
         return [], [], []
 
-    if last_pivot_pos + depth >= n:
+    if n - last_pivot_pos - 1 < depth:
         return pivots_indices, pivots_values, pivots_directions
 
     for i in range(last_pivot_pos + 1, n):

diff --git a/quickadapter/user_data/strategies/Utils.py b/quickadapter/user_data/strategies/Utils.py
index 8cb61049425c2004d20a0db8a5940a99dbd89dd9..0634c6266285c888be17c53d32feb769db9c74ea 100644 (file)
--- a/quickadapter/user_data/strategies/Utils.py
+++ b/quickadapter/user_data/strategies/Utils.py
@@ -451,7 +451,7 @@ def zigzag(
     else:
         return [], [], []
 
-    if last_pivot_pos + depth >= n:
+    if n - last_pivot_pos - 1 < depth:
         return pivots_indices, pivots_values, pivots_directions
 
     for i in range(last_pivot_pos + 1, n):