From a40d66bb2275641d5c31ad5095fb982e7bfd3feb Mon Sep 17 00:00:00 2001 From: Jan Kieseler Date: Sun, 15 Sep 2024 22:17:50 +0200 Subject: [PATCH] for dependencies --- bind/G4Calo.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/bind/G4Calo.py b/bind/G4Calo.py index a54ce6a..2b2843d 100644 --- a/bind/G4Calo.py +++ b/bind/G4Calo.py @@ -211,13 +211,13 @@ def run_batch( seed += 1 print(f"Batch seed: {seed}") - if no_mp: - dfs = [] - for i in range(nCores): - dfs.append(_run_mini_batch_silent(gd, nevents[i], particleSpec, minEnergy_GeV, maxEnergy_GeV, seed, i)) - else: - with multiprocessing.Pool(nCores) as pool: - dfs = pool.starmap(_run_mini_batch_silent, [(gd, nevents[i], particleSpec, minEnergy_GeV, maxEnergy_GeV, seed, i) for i in range(nCores)]) + #if no_mp: + # dfs = [] + # for i in range(nCores): + # dfs.append(_run_mini_batch_silent(gd, nevents[i], particleSpec, minEnergy_GeV, maxEnergy_GeV, seed, i)) + #else: + with multiprocessing.Pool(nCores) as pool: + dfs = pool.starmap(_run_mini_batch_silent, [(gd, nevents[i], particleSpec, minEnergy_GeV, maxEnergy_GeV, seed, i) for i in range(nCores)]) alldf = pd.concat(dfs) alldf.reset_index(drop=True, inplace=True)