changed yaml

msmbuilder · Apr 17, 2022 · 9e18e6a · 9e18e6a
1 parent af38cb3
commit 9e18e6a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/msmbuilder/io/project_template.py b/msmbuilder/io/project_template.py
@@ -151,8 +151,14 @@ def get_source(self, environment, template):
             return source, filename, uptodate
 
         end = source[beg:].find(end_str) + beg
+
+        # yaml update
+        try:
+            self.meta[filename] = yaml.load(source[beg + len(beg_str):end])
+        except:
+            from yaml.loader import Loader
+            self.meta[filename] = yaml.load(source[beg + len(beg_str):end],Loader=Loader)
 
-        self.meta[filename] = yaml.load(source[beg + len(beg_str):end])
         remove_meta = source[:beg] + source[end:]
         return remove_meta, filename, uptodate