Update parallelize.py

use `.lower()` to convert possible "upper" keywords to "lower"
IncubatorShokuhou · Oct 13, 2020 · 42f0c08 · 42f0c08
1 parent d0ee6e4
commit 42f0c08
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/parallelize.py b/parallelize.py
@@ -9,11 +9,11 @@ def get_per_cpu(seq,n_jobs= "half"):
     elif isinstance(n_jobs,float):
         n_jobs_in = int(n_jobs)
     elif isinstance(n_jobs,str):
-        if n_jobs == "all":
+        if n_jobs.lower() == "all":
             n_jobs_in = multiprocessing.cpu_count()
-        elif n_jobs == "half":
+        elif n_jobs.lower() == "half":
             n_jobs_in = int( multiprocessing.cpu_count() / 2 )
-        elif n_jobs == "quarter":
+        elif n_jobs.lower() == "quarter":
             n_jobs_in = int( multiprocessing.cpu_count() / 4 )
         else:
             try: