lars/minicalosim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

for dependencies

Browse Source
This commit is contained in:
Jan Kieseler
2024-09-15 22:17:50 +02:00
parent 854402320c
commit a40d66bb22
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bind/G4Calo.py
+7 -7
View File
@@ -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)