1 files changed, 5 insertions, 1 deletions
diff --git a/utils/submit-grid-jobs b/utils/submit-grid-jobs
index 13bddfe..37b3303 100755
--- a/utils/submit-grid-jobs
+++ b/utils/submit-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
 import string
 from os.path import split, splitext, join, abspath
 import uuid
@@ -127,7 +131,7 @@ if __name__ == '__main__':
         else:
             output = check_output([join(args.dir,"zdab-cat"),filename])
 
-        data = yaml.load(output,Loader=yaml.loader.SafeLoader)
+        data = yaml.load(output,Loader=Loader)
 
         filename = abspath(filename)