Add long description and version in setup.py.

PiperOrigin-RevId: 302047654
google-deepmind · Apr 23, 2020 · 4aa02b3 · 4aa02b3
1 parent d251b24
commit 4aa02b3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/rlax/__init__.py b/rlax/__init__.py
@@ -80,6 +80,8 @@
 from rlax._src.value_learning import vtrace
 from rlax._src.vtrace import vtrace_td_error_and_advantage
 
+__version__ = "0.0.1"
+
 __all__ = (
     "add_gaussian_noise",
     "add_ornstein_uhlenbeck_noise",

diff --git a/setup.py b/setup.py
@@ -18,12 +18,26 @@
 from setuptools import find_namespace_packages
 from setuptools import setup
 
+
+def _get_version():
+  with open('rlax/__init__.py') as fp:
+    for line in fp:
+      if line.startswith('__version__'):
+        g = {}
+        exec(line, g)  # pylint: disable=exec-used
+        return g['__version__']
+    raise ValueError('`__version__` not defined in `rlax/__init__.py`')
+
+
 setup(
     name='rlax',
+    version=_get_version(),
     url='https://github.com/deepmind/rlax',
     license='Apache 2.0',
     author='DeepMind',
     description=('A library of reinforcement learning building blocks in JAX.'),
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
     author_email='rlax-dev@google.com',
     keywords='reinforcement-learning python machine learning',
     packages=find_namespace_packages(exclude=['*_test.py']),