diff options
Diffstat (limited to 'utils/chi2')
| -rwxr-xr-x | utils/chi2 | 7 |
1 files changed, 3 insertions, 4 deletions
@@ -487,7 +487,6 @@ if __name__ == '__main__': parser.add_argument("--save", action='store_true', default=False, help="save corner plots for backgrounds") parser.add_argument("--mc", nargs='+', required=True, help="atmospheric MC files") parser.add_argument("--muon-mc", nargs='+', required=True, help="muon MC files") - parser.add_argument("--nhit-thresh", type=int, default=None, help="nhit threshold to apply to events before processing (should only be used for testing to speed things up)") parser.add_argument("--steps", type=int, default=1000, help="number of steps in the MCMC chain") parser.add_argument("--multinomial-prob-size", type=int, default=10000, help="number of p values to compute") parser.add_argument("--coverage", type=int, default=0, help="plot p value coverage") @@ -513,7 +512,7 @@ if __name__ == '__main__': evs = [] for run, df in rhdr.groupby('run'): - evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh)) + evs.append(get_events(df.filename.values, merge_fits=True)) ev = pd.concat(evs).reset_index() ev = correct_energy_bias(ev) @@ -524,14 +523,14 @@ if __name__ == '__main__': # and otherwise the GTIDs will clash ev_mcs = [] for filename in args.mc: - ev_mcs.append(get_events([filename], merge_fits=True, nhit_thresh=args.nhit_thresh, mc=True)) + ev_mcs.append(get_events([filename], merge_fits=True, mc=True)) ev_mc = pd.concat([ev_mc for ev_mc in ev_mcs if len(ev_mc) > 0]).reset_index() if (~rhdr.run.isin(ev_mc.run)).any(): print_warning("Error! The following runs have no Monte Carlo: %s" % \ np.unique(rhdr.run[~rhdr.run.isin(ev_mc.run)].values)) - muon_mc = get_events(args.muon_mc, merge_fits=True, nhit_thresh=args.nhit_thresh, mc=True) + muon_mc = get_events(args.muon_mc, merge_fits=True, mc=True) weights = pd.concat([read_hdf(filename, "weights") for filename in args.weights],ignore_index=True) # The next two things we have to do are reweight the Monte Carlo data since |