bug in pool.map

Change-Id: Iac637395ea72e4296e1ee5d34256ca9988799180
BlueBrain · Feb 18, 2020 · ba99578 · ba99578
1 parent ba116dd
commit ba99578
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/diameter_synthesis/build_diameters.py b/diameter_synthesis/build_diameters.py
@@ -85,10 +85,10 @@ def build_diameters(morphologies_dict, models_params, config):
 
         # generate diameters in parallel
         worker = Worker(model, models_params, config)
-        pool = multiprocessing.Pool(config["n_cpu"])
-        if config["n_cpu"]:
+        if config["n_cpu"] == 1:
             mapping = map
         else:
+            pool = multiprocessing.Pool(config["n_cpu"])
             mapping = pool.map
 
         list(mapping(worker, neurons))