Fix bug in neutron_plots.py (#26)

* Update neutron_plots.py

import uproot as ur

* fixed another bug in the neutron_plots.py

* Update config.yml

Make sure the neutron::analyze stage actually runs.

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 @@ def gauss(x, A,mu, sigma):
     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

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