Replace if else with a minimum comparison

sktime · Oct 25, 2023 · afde81e · afde81e
1 parent b89db4a
commit afde81e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sktime/transformations/series/_clasp_numba.py b/sktime/transformations/series/_clasp_numba.py
@@ -111,10 +111,8 @@ def _compute_distances_iterative(X, m, k):
         )
         dist[trivialMatchRange[0] : trivialMatchRange[1]] = np.inf
 
-        if k >= len(dist):
-            idx = np.argpartition(dist, len(dist) - 1)
-        else:
-            idx = np.argpartition(dist, k)
+        _k = min(k, len(dist) - 1)
+        idx = np.argpartition(dist, _k)
 
         knns[order, :] = idx[:k]
         dot_prev = dot_rolled