diff options
Diffstat (limited to 'utils/chi2')
| -rwxr-xr-x | utils/chi2 | 5 |
1 files changed, 5 insertions, 0 deletions
@@ -431,6 +431,7 @@ if __name__ == '__main__': parser.add_argument("--print-nll", action='store_true', default=False, help="print nll values") parser.add_argument("--walkers", type=int, default=100, help="number of walkers") parser.add_argument("--thin", type=int, default=10, help="number of steps to thin") + parser.add_argument("--run-list", default=None, help="run list") args = parser.parse_args() setup_matplotlib(args.save) @@ -444,6 +445,10 @@ if __name__ == '__main__': evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh)) ev = pd.concat(evs) + if args.run_list is not None: + runs = np.genfromtxt(args.run_list) + ev = ev[ev.run.isin(runs)] + ev = correct_energy_bias(ev) # Note: We loop over the MC filenames here instead of just passing the |