use yaml.loader.SafeLoader

1 files changed, 2 insertions, 5 deletions

diff --git a/utils/submit-grid-jobs b/utils/submit-grid-jobs
index 06ddae9..13bddfe 100755
--- a/utils/submit-grid-jobs
+++ b/utils/submit-grid-jobs
@@ -15,10 +15,7 @@
 # this program. If not, see <https://www.gnu.org/licenses/>.
 
 from __future__ import print_function, division
-try:
-    from yaml import full_load as load
-except ImportError:
-    from yaml import load
+import yaml
 import string
 from os.path import split, splitext, join, abspath
 import uuid
@@ -130,7 +127,7 @@ if __name__ == '__main__':
         else:
             output = check_output([join(args.dir,"zdab-cat"),filename])
 
-        data = load(output)
+        data = yaml.load(output,Loader=yaml.loader.SafeLoader)
 
         filename = abspath(filename)