Removed variadic args to support numba on python < 3.10

Author: lucasimi

src/tdamapper/utils/quickselect.py

@@ -16,26 +16,26 @@ def swap_all(arr, i, j, extra1=None, extra2=None):
 
 
 @njit
-def partition(data, start, end, p_ord, *extra):
+def partition(data, start, end, p_ord, extra1=None, extra2=None):
     higher = start
     for j in range(start, end):
         j_ord = data[j]
         if j_ord < p_ord:
-            swap_all(data, higher, j, *extra)
+            swap_all(data, higher, j, extra1, extra2)
             higher += 1
     return higher
 
 
 @njit
-def quickselect(data, start, end, k, *extra):
+def quickselect(data, start, end, k, extra1=None, extra2=None):
     if (k < start) or (k >= end):
         return
     start_, end_, higher = start, end, None
     while higher != k + 1:
         p = data[k]
-        swap_all(data, start_, k, *extra)
-        higher = partition(data, start_ + 1, end_, p, *extra)
-        swap_all(data, start_, higher - 1, *extra)
+        swap_all(data, start_, k, extra1, extra2)
+        higher = partition(data, start_ + 1, end_, p, extra1, extra2)
+        swap_all(data, start_, higher - 1, extra1, extra2)
         if k <= higher - 1:
             end_ = higher
         else: