Refactor version string

Fixes #163
inferno-pytorch · Jan 23, 2019 · b0ed9bd · b0ed9bd
1 parent 955d462
commit b0ed9bd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/inferno/__init__.py b/inferno/__init__.py
@@ -6,9 +6,10 @@
 from . import io
 from . import trainers
 from . import utils
+from .version import __version__
+
 
 __all__ = ['extensions', 'io', 'trainers', 'utils']
 
 __author__ = """Nasim Rahaman"""
 __email__ = 'nasim.rahaman@iwr.uni-heidelberg.de'
-__version__ = '0.3.1'
diff --git a/inferno/version.py b/inferno/version.py
@@ -0,0 +1 @@
+__version__ = '0.3.1'
diff --git a/setup.py b/setup.py
@@ -4,6 +4,7 @@
 """The setup script."""
 
 from setuptools import setup, find_packages
+from inferno import __version__
 
 with open('README.rst') as readme_file:
     readme = readme_file.read()
@@ -38,7 +39,7 @@
 
 setup(
     name='inferno-pytorch',
-    version='0.3.1',
+    version=__version__,
     description="Inferno is a little library providing utilities and convenience functions/classes around PyTorch.",
     long_description=readme + '\n\n' + history,
     author="Nasim Rahaman",