diff options
Diffstat (limited to 'utils/cat-grid-jobs')
| -rwxr-xr-x | utils/cat-grid-jobs | 6 |
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 |