np.arange: do not pass start as kwarg

saezlab · Jun 17, 2024 · a1bf4a7 · a1bf4a7
1 parent 1a52b33
commit a1bf4a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/decoupler/method_aucell.py b/decoupler/method_aucell.py
@@ -34,7 +34,7 @@ def nb_aucell(row, net, starts, offsets, n_up, n_fsets):
         fset = net[srt:off]
 
         # Compute max AUC for fset
-        x_th = np.arange(start=1, stop=fset.shape[0]+1, dtype=nb.i8)
+        x_th = np.arange(1, stop=fset.shape[0]+1, dtype=nb.i8)
         x_th = x_th[x_th < n_up]
         max_auc = np.sum(np.diff(np.append(x_th, n_up)) * x_th)
 

diff --git a/decoupler/method_gsva.py b/decoupler/method_gsva.py
@@ -83,7 +83,7 @@ def density(mat, kcdf=False):
 @nb.njit(nb.types.Tuple((nb.f4[:, :], nb.i8[:, :]))(nb.f4[:, :]), parallel=True, cache=True)
 def nb_get_D_I(mat):
     n = mat.shape[1]
-    rev_idx = np.abs(np.arange(start=n, stop=0, step=-1, dtype=nb.f4) - n / 2)
+    rev_idx = np.abs(np.arange(n, stop=0, step=-1, dtype=nb.f4) - n / 2)
     Idx = np.zeros(mat.shape, dtype=nb.i8)
     for i in nb.prange(mat.shape[0]):
         Idx[i] = np.argsort(-mat[i])