From b00dfd1a8d55db70e85542d04f14aa3dcd0f36b9 Mon Sep 17 00:00:00 2001 From: Sherman Siu Date: Sat, 27 Apr 2024 05:17:04 -0400 Subject: [PATCH] Allow default number of nb_workers to be configured with "None". --- pandarallel/core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pandarallel/core.py b/pandarallel/core.py index c54ec14..73110b3 100644 --- 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()