cleaning up

Author: eamonnfaherty

cloudformation_docs/sdk.py

@@ -1,12 +1,13 @@
 from . import core
 import json
-from cfn_flip import yaml
+from cfn_tools import load_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.load_yaml(template_content), name)
+    result = core.generate(load_yaml(template_content), name)
     return result

setup.py

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