Allow default number of nb_workers to be configured with "None".

nalepae · Apr 27, 2024 · b00dfd1 · b00dfd1
1 parent 261a652
commit b00dfd1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pandarallel/core.py b/pandarallel/core.py
@@ -453,11 +453,14 @@ class pandarallel:
     def initialize(
         cls,
         shm_size_mb=None,
-        nb_workers=NB_PHYSICAL_CORES,
+        nb_workers: Optional[int] = None,
         progress_bar=False,
         verbose=2,
         use_memory_fs: Optional[bool] = None,
     ) -> None:
+        if nb_workers is None:
+            nb_workers = NB_PHYSICAL_CORES
+
         show_progress_bars = progress_bar
         is_memory_fs_available = Path(MEMORY_FS_ROOT).exists()