add __version__

pyro-ppl · May 31, 2019 · 474fc41 · 474fc41
1 parent b6e7f7c
commit 474fc41
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/numpyro/__init__.py b/numpyro/__init__.py
@@ -1 +1,2 @@
 import numpyro.patch  # noqa: F401
+from numpyro.version import __version__
diff --git a/numpyro/version.py b/numpyro/version.py
@@ -0,0 +1 @@
+__version__ = '0.1.0'
diff --git a/setup.py b/setup.py
@@ -1,9 +1,16 @@
 from __future__ import absolute_import, division, print_function
 
+import os
 import sys
 
 from setuptools import find_packages, setup
 
+PROJECT_PATH = os.path.dirname(os.path.abspath(__file__))
+
+# Find version
+for line in open(os.path.join(PROJECT_PATH, 'numpyro', 'version.py')):
+    if line.startswith('__version__ = '):
+        version = line.strip().split()[2][1:-1]
 
 # READ README.md for long description on PyPi.
 try: