fixing yaml load issue

Author: eamonnfaherty

cloudformation_docs/sdk.py

@@ -1,13 +1,12 @@
 from . import core
 import json
-import yaml
-
+from cfn_flip import yaml
 
 def generate_from_json(template_content, name):
     result = core.generate(json.loads(template_content), name)
     return result
 
 
 def generate_from_yaml(template_content, name):
-    result = core.generate(yaml.safe_load(template_content), name)
+    result = core.generate(yaml.load_yaml(template_content), name)
     return result

setup.py

@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="cloudformation-docs",
-    version="0.3.0",
+    version="0.3.1",
     author="Eamonn Faherty",
     author_email="packages@designandsolve.co.uk",
     description="Generate docs from cloudformation templates",