Update tau_plots.py

read from the correct file locations
eic · Oct 25, 2024 · 906de21 · 906de21
1 parent 5ec80d0
commit 906de21
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/benchmarks/insert_tau/analysis/tau_plots.py b/benchmarks/insert_tau/analysis/tau_plots.py
@@ -17,7 +17,14 @@
     pass
 
 import uproot as ur
-arrays_sim={p:ur.open(f'sim_output/insert_tau/{config}_rec_tau-_{p}GeV.edm4eic.root:events').arrays() for p in (20, 30, 40, 50, 60,80,100)}
+arrays_sim={}
+momenta=20, 30, 40, 50, 60,80,100
+for p in momenta:
+    arrays_sim[p] = ur.concatenate({
+        f'sim_output/insert_tau/{config}_rec_tau-_{p}GeV_{index}.edm4eic.root': 'events'
+        for index in range(5)
+    })
+
 
 for a in arrays_sim.values():
     #recon