From 94f138d68338e49a94a1518ea4b554ea37d16099 Mon Sep 17 00:00:00 2001
From: tlatorre <tlatorre@uchicago.edu>
Date: Wed, 5 Jun 2019 15:41:20 -0400
Subject: try to import CLoader if possible since it's *much* faster

---
 utils/cat-grid-jobs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'utils/cat-grid-jobs')

diff --git a/utils/cat-grid-jobs b/utils/cat-grid-jobs
index 43fef50..84623b8 100755
--- a/utils/cat-grid-jobs
+++ b/utils/cat-grid-jobs
@@ -16,6 +16,10 @@
 
 from __future__ import print_function, division
 import yaml
+try:
+    from yaml import CLoader as Loader
+except ImportError:
+    from yaml.loader import SafeLoader as Loader
 
 if __name__ == '__main__':
     import argparse
@@ -30,7 +34,7 @@ if __name__ == '__main__':
 
     for filename in args.filenames:
         with open(filename) as f:
-            data = yaml.load(f.read(),Loader=yaml.loader.SafeLoader)
+            data = yaml.load(f.read(),Loader=Loader)
 
         if data is None:
             continue
-- 
cgit