Unbroke the stuff I broke in 0.4.1

tehsmyers · Aug 17, 2014 · 8aa4b64 · 8aa4b64
1 parent bb7518f
commit 8aa4b64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/cfn_pyplates/cli.py b/cfn_pyplates/cli.py
@@ -20,7 +20,7 @@
 from schema import Schema, Use, Or
 import yaml
 
-from cfn_pyplates import __version__, core, functions
+from cfn_pyplates import core, functions
 from cfn_pyplates.exceptions import Error
 from cfn_pyplates.options import OptionsMapping
 
@@ -94,7 +94,9 @@ def generate():
 
   Be careful.
 """
-    args = docopt(generate.__doc__, version=__version__)
+    from pkg_resources import require
+    version = require("cfn-pyplates")[0].version
+    args = docopt(generate.__doc__, version=version)
     scheme = Schema({
         '<pyplate>': Use(open),
         '<outfile>': Or(None, '-', Use(_open_writable)),

diff --git a/tests/test_version.py b/tests/test_version.py