diff options
| -rwxr-xr-x | utils/cat-grid-jobs | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/utils/cat-grid-jobs b/utils/cat-grid-jobs index 83a8004..1f83bea 100755 --- a/utils/cat-grid-jobs +++ b/utils/cat-grid-jobs @@ -130,10 +130,11 @@ if __name__ == '__main__': import os import sys import h5py + import glob parser = argparse.ArgumentParser("concatenate fit results from grid jobs into a single file") parser.add_argument("zdab", help="zdab input file") - parser.add_argument("filenames", nargs='+', help="input files") + parser.add_argument("directory", help="directory with grid results") parser.add_argument("-o", "--output", type=str, help="output filename", required=True) args = parser.parse_args() @@ -158,7 +159,7 @@ if __name__ == '__main__': with h5py.File(args.output,"a") as fout: total_events = fout['ev'].shape[0] - for filename in args.filenames: + for filename in glob.glob(join(args.directory,'*.hdf5')): with h5py.File(filename) as f: # Check to see if the git sha1 match if fout.attrs['git_sha1'] != f.attrs['git_sha1']: |