From ba995789335547c460e70772282beed3b3707a21 Mon Sep 17 00:00:00 2001 From: Alexis Arnaudon Date: Tue, 18 Feb 2020 11:32:29 +0100 Subject: [PATCH] bug in pool.map Change-Id: Iac637395ea72e4296e1ee5d34256ca9988799180 --- diameter_synthesis/build_diameters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diameter_synthesis/build_diameters.py b/diameter_synthesis/build_diameters.py index 9d29c1d..011c26b 100644 --- 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))