update cat-grid-jobs to sort output based on gtid

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))