diff options
| -rwxr-xr-x | utils/cat-grid-jobs | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/utils/cat-grid-jobs b/utils/cat-grid-jobs index d43dfad..43fef50 100755 --- a/utils/cat-grid-jobs +++ b/utils/cat-grid-jobs @@ -22,7 +22,7 @@ if __name__ == '__main__': import matplotlib.pyplot as plt import numpy as np - parser = argparse.ArgumentParser("plot fit results") + parser = argparse.ArgumentParser("concatenate fit results from grid jobs into a single file") parser.add_argument("filenames", nargs='+', help="input files") args = parser.parse_args() @@ -30,13 +30,16 @@ if __name__ == '__main__': for filename in args.filenames: with open(filename) as f: - data = yaml.load(f.read()) + data = yaml.load(f.read(),Loader=yaml.loader.SafeLoader) if data is None: continue for event in data['data']: if event['ev'] is not None: + # if the ev branch is filled in, it means the event was fit cat.append(event) + cat = sorted(cat,key=lambda event: event['ev'][0]['gtid']) + print(yaml.dump({'data':cat},default_flow_style=False)) |