aboutsummaryrefslogtreecommitdiff
path: root/utils/cat-grid-jobs
diff options
context:
space:
mode:
authortlatorre <tlatorre@uchicago.edu>2019-06-05 15:41:20 -0400
committertlatorre <tlatorre@uchicago.edu>2019-06-05 15:41:20 -0400
commit94f138d68338e49a94a1518ea4b554ea37d16099 (patch)
tree20d0514d4f7d6b96ca7e46f699024980f053ee03 /utils/cat-grid-jobs
parent74c64892d70c2da37954653be7a85dc9efc71f1a (diff)
downloadsddm-94f138d68338e49a94a1518ea4b554ea37d16099.tar.gz
sddm-94f138d68338e49a94a1518ea4b554ea37d16099.tar.bz2
sddm-94f138d68338e49a94a1518ea4b554ea37d16099.zip
try to import CLoader if possible since it's *much* faster
Diffstat (limited to 'utils/cat-grid-jobs')
-rwxr-xr-xutils/cat-grid-jobs6
1 files changed, 5 insertions, 1 deletions
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