more places to use yaml.safe_load

This commit is contained in:
Michael DeHaan 2013-02-23 13:34:14 -05:00
parent a9ea3136cf
commit b75b9681a8
4 changed files with 4 additions and 4 deletions

View file

@ -45,7 +45,7 @@ def get_docstring(filename, verbose=False):
for child in M.body:
if isinstance(child, ast.Assign):
if 'DOCUMENTATION' in (t.id for t in child.targets):
doc = yaml.load(child.value.s)
doc = yaml.safe_load(child.value.s)
if 'EXAMPLES' in (t.id for t in child.targets):
plainexamples = child.value.s[1:] # Skip first empty line
except: