avoid passing unncessary argument k

Author: NimaSarajpoor

stumpy/gpu_aamp.py

@@ -42,7 +42,6 @@ def _compute_and_update_PI_kernel(
     compute_p_norm,
     bfs,
     nlevel,
-    k,
 ):
     """
     A Numba CUDA kernel to update the non-normalized (i.e., without z-normalization)
@@ -115,11 +114,6 @@ def _compute_and_update_PI_kernel(
     compute_p_norm : bool
         A boolean flag for whether or not to compute the p-norm
 
-    k : int
-        The number of top `k` smallest distances used to construct the matrix profile.
-        Note that this will increase the total computational time and memory usage
-        when k > 1.
-
     Returns
     -------
     None
@@ -369,7 +363,6 @@ def _gpu_aamp(
             False,
             device_bfs,
             nlevel,
-            k,
         )
 
         for i in range(range_start, range_stop):
@@ -396,7 +389,6 @@ def _gpu_aamp(
                 True,
                 device_bfs,
                 nlevel,
-                k,
             )
 
         profile = device_profile.copy_to_host()

stumpy/gpu_stump.py

@@ -18,7 +18,7 @@
 @cuda.jit(
     "(i8, f8[:], f8[:], i8,  f8[:], f8[:], f8[:], f8[:], f8[:],"
     "f8[:], f8[:], i8, b1, i8, f8[:, :], f8[:], f8[:], i8[:, :], i8[:], i8[:],"
-    "b1, i8[:], i8, i8)"
+    "b1, i8[:], i8)"
 )
 def _compute_and_update_PI_kernel(
     i,
@@ -44,7 +44,6 @@ def _compute_and_update_PI_kernel(
     compute_QT,
     bfs,
     nlevel,
-    k,
 ):
     """
     A Numba CUDA kernel to update the matrix profile and matrix profile indices
@@ -127,11 +126,6 @@ def _compute_and_update_PI_kernel(
         The number of levels in the binary search tree from which the array
         `bfs` is obtained.
 
-    k : int
-        The number of top `k` smallest distances used to construct the matrix profile.
-        Note that this will increase the total computational time and memory usage
-        when k > 1.
-
     Returns
     -------
     None
@@ -400,7 +394,6 @@ def _gpu_stump(
             False,
             device_bfs,
             nlevel,
-            k,
         )
 
         for i in range(range_start, range_stop):
@@ -428,7 +421,6 @@ def _gpu_stump(
                 True,
                 device_bfs,
                 nlevel,
-                k,
             )
 
         profile = device_profile.copy_to_host()

stumpy/stump.py

@@ -142,11 +142,6 @@ def _compute_diagonal(
         Set to `True` if this is a self-join. Otherwise, for AB-join, set this to
         `False`. Default is `True`.
 
-    k : int
-        The number of top `k` smallest distances used to construct the matrix profile.
-        Note that this will increase the total computational time and memory usage
-        when k > 1.
-
     Returns
     -------
     None