diff options
Diffstat (limited to 'utils')
| -rwxr-xr-x | utils/chi2 | 5 | ||||
| -rwxr-xr-x | utils/dm-search | 5 |
2 files changed, 10 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 diff --git a/utils/dm-search b/utils/dm-search index 17f107a..6c78d4e 100755 --- a/utils/dm-search +++ b/utils/dm-search @@ -481,6 +481,7 @@ if __name__ == '__main__': 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("--test", type=int, default=0, help="run tests to check discovery threshold") + parser.add_argument("--run-list", default=None, help="run list") args = parser.parse_args() setup_matplotlib(args.save) @@ -494,6 +495,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)] + livetime = 0.0 livetime_pulse_gt = 0.0 for ev in evs: |