From 9e18e6adddb5155fef897ea6249896bcea896fae Mon Sep 17 00:00:00 2001 From: bojunliu0818 Date: Sun, 17 Apr 2022 11:33:30 -0500 Subject: [PATCH] changed yaml --- msmbuilder/io/project_template.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/msmbuilder/io/project_template.py b/msmbuilder/io/project_template.py index ead378f7..b86732e6 100644 --- 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