Merge pull request #22 from uweschmitt/fix_install

fix installation when six is not installed yet
adrn · May 2, 2019 · e00242b · e00242b
2 parents d2538b7 + 4a995b8
commit e00242b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/schwimmbad/__init__.py b/schwimmbad/__init__.py
@@ -12,8 +12,9 @@
     - SerialPool: A serial pool, which uses the built-in ``map`` function
 
 """
+import pkg_resources
 
-__version__ = "0.4.dev"
+__version__ = pkg_resources.require(__package__)[0].version
 __author__ = "Adrian Price-Whelan <adrianmpw@gmail.com>"
 
 # Standard library

diff --git a/setup.py b/setup.py
@@ -4,10 +4,10 @@
 # python setup.py sdist upload
 
 # Standard library
-import sys
 import locale
 import os
 import subprocess
+import sys
 import warnings
 
 try:
@@ -31,8 +31,7 @@ def rd(filename):
         return r
 
 # Remove the .dev for release
-import schwimmbad
-VERSION = schwimmbad.__version__
+VERSION = "0.4.dev"
 
 # Indicates if this version is a release version
 RELEASE = 'dev' not in VERSION