Improve handling of n_jobs=-1

songlab-cal · Dec 6, 2022 · 5b19f4b · 5b19f4b
1 parent 6184fd5
commit 5b19f4b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
diff --git a/differential_splicing_example.ipynb b/differential_splicing_example.ipynb
diff --git a/scquint/differential_splicing.py b/scquint/differential_splicing.py
@@ -208,7 +208,7 @@ def _run_differential_splicing(
 
     X = adata.X.toarray()  # for easier parallelization using Python's libraries
 
-    if n_jobs is not None and n_jobs > 1:
+    if n_jobs is not None and n_jobs != 1:
         dfs_intron_group, dfs_intron = zip(
             *Parallel(n_jobs=n_jobs)(
                 delayed(run_regression)((c, X[:, intron_group_introns[c]], cell_idx_a, cell_idx_b))