|
1 | 1 | from __future__ import absolute_import, division, print_function |
2 | 2 |
|
| 3 | +import os |
| 4 | +import sys |
| 5 | + |
3 | 6 | from setuptools import find_packages, setup |
4 | 7 |
|
| 8 | +PROJECT_PATH = os.path.dirname(os.path.abspath(__file__)) |
| 9 | + |
| 10 | +# Find version |
| 11 | +for line in open(os.path.join(PROJECT_PATH, 'numpyro', 'version.py')): |
| 12 | + if line.startswith('__version__ = '): |
| 13 | + version = line.strip().split()[2][1:-1] |
| 14 | + |
| 15 | +# READ README.md for long description on PyPi. |
| 16 | +try: |
| 17 | + long_description = open('README.md', encoding='utf-8').read() |
| 18 | +except Exception as e: |
| 19 | + sys.stderr.write('Failed to convert README.md to rst:\n {}\n'.format(e)) |
| 20 | + sys.stderr.flush() |
| 21 | + long_description = '' |
| 22 | + |
| 23 | + |
5 | 24 | setup( |
6 | 25 | name='numpyro', |
7 | | - version='0.0.0', |
| 26 | + version='0.1.0', |
8 | 27 | description='Pyro PPL on Numpy', |
9 | 28 | packages=find_packages(include=['numpyro', 'numpyro.*']), |
10 | 29 | url='https://github.com/neerajprad/numpyro', |
11 | 30 | author='Uber AI Labs', |
12 | 31 | |
13 | 32 | install_requires=[ |
14 | | - 'jax>=0.1.35', |
| 33 | + # TODO: Remove soon as JAX's API becomes stable |
| 34 | + 'jax==0.1.35', |
15 | 35 | 'jaxlib>=0.1.14', |
16 | 36 | 'tqdm', |
17 | 37 | ], |
|
21 | 41 | 'dev': ['ipython'], |
22 | 42 | 'examples': ['matplotlib'], |
23 | 43 | }, |
| 44 | + long_description=long_description, |
| 45 | + long_description_content_type='text/markdown', |
24 | 46 | tests_require=['flake8', 'pytest>=4.1'], |
25 | 47 | keywords='probabilistic machine learning bayesian statistics', |
26 | 48 | license='MIT License', |
|
0 commit comments