diff --git a/benchmarks/neutron/analysis/neutron_plots.py b/benchmarks/neutron/analysis/neutron_plots.py index a7b919b690660d4a4bd77010c7fc49d3d83a806a..41583b9c9e592e96b2aee498cbda9e8a14355b2d 100644 --- a/benchmarks/neutron/analysis/neutron_plots.py +++ b/benchmarks/neutron/analysis/neutron_plots.py @@ -1,4 +1,4 @@ -import numpy as np, pandas as pd, matplotlib.pyplot as plt, matplotlib as mpl, awkward as ak, sys +import numpy as np, pandas as pd, matplotlib.pyplot as plt, matplotlib as mpl, awkward as ak, sys, uproot as ur import mplhep as hep hep.style.use("CMS") @@ -57,7 +57,7 @@ for array in arrays_sim.values(): w=E array['theta_recon']=np.sum(np.arccos(zp/r)*w, axis=-1)/np.sum(w, axis=-1) - array['eta_recon']=-np.log(np.tan(['theta_recon']/2)) + array['eta_recon']=-np.log(np.tan(array['theta_recon']/2)) array['E_Hcal']=np.sum(array['HcalEndcapPInsertClusters.energy'], axis=-1)#*20/12.5 diff --git a/benchmarks/neutron/config.yml b/benchmarks/neutron/config.yml index be42df42a217b9aa56ab92e9f851865be7c8e3d1..304b8311a47eac6ed00782815936edb5b047a9d9 100644 --- a/benchmarks/neutron/config.yml +++ b/benchmarks/neutron/config.yml @@ -24,7 +24,8 @@ neutron:analyze: extends: .phy_benchmark needs: ["neutron:simulate"] script: - - snakemake --cores 1 --touch results/epic_craterlake/neutron + - mkdir -p results/epic_craterlake + - python benchmarks/neutron/analysis/neutron_plots.py results/epic_craterlake/neutron neutron:results: stage: collect