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)