🎨 accept n_jobs=-1

ThomasParistech · Jun 5, 2024 · f001fd7 · f001fd7
1 parent 88d0513
commit f001fd7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/dobble/card.py b/dobble/card.py
@@ -388,4 +388,4 @@ def main(masks_folder: str,
                    for card_idx, (symbols, scale_targets) in enumerate(zip(cards, scale_targets_per_card))]
 
     multiprocess(generate_card, list_kwargs, tqdm_title="Generate Cards",
-                 n_jobs=1 if DEBUG or DEBUG_FINAL else None)
+                 n_jobs=1 if DEBUG or DEBUG_FINAL else -1)
diff --git a/dobble/utils.py b/dobble/utils.py
@@ -94,6 +94,8 @@ def multiprocess(process_func: Callable[..., None],
 
     if n_jobs is None:
         n_jobs = math.floor(0.8 * cpu_count())
+    elif n_jobs < 0:
+        n_jobs = cpu_count()
 
     n_jobs = min(n_jobs, cpu_count())