From 491cd9317b0c215adf081d1ea77b34fadc131c90 Mon Sep 17 00:00:00 2001 From: tlatorre Date: Sun, 2 Jun 2019 15:48:27 -0400 Subject: update cat-grid-jobs to sort output based on gtid --- utils/cat-grid-jobs | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'utils/cat-grid-jobs') 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)) -- cgit