Project import generated by Copybara.

PiperOrigin-RevId: 383644761
google-deepmind · Jul 8, 2021 · af776aa · af776aa
1 parent 3566d85
commit af776aa
Show file tree

Hide file tree

Showing 8 changed files with 476 additions and 10 deletions.
diff --git a/.pylintrc b/.pylintrc
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1,4 @@
+include README.md
+include LICENSE
+include requirements/*
+include rlax/py.typed
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -0,0 +1 @@
+# Placeholder
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -0,0 +1,5 @@
+absl-py>=0.9.0
+chex>=0.0.8
+jax>=0.1.55
+jaxlib>=0.1.37
+numpy>=1.18.0
diff --git a/requirements/requirements_examples.txt b/requirements/requirements_examples.txt
@@ -0,0 +1 @@
+dm-env>=1.2
diff --git a/rlax/_src/__init__.py b/rlax/_src/__init__.py
@@ -0,0 +1,14 @@
+# Copyright 2021 DeepMind Technologies Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
diff --git a/rlax/py.typed b/rlax/py.typed
diff --git a/setup.py b/setup.py
@@ -14,9 +14,12 @@
 # ==============================================================================
 """Install script for setuptools."""
 
+import os
 from setuptools import find_namespace_packages
 from setuptools import setup
 
+_CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
+
 
 def _get_version():
   with open('rlax/__init__.py') as fp:
@@ -28,29 +31,36 @@ def _get_version():
     raise ValueError('`__version__` not defined in `rlax/__init__.py`')
 
 
+def _parse_requirements(path):
+
+  with open(os.path.join(_CURRENT_DIR, path)) as f:
+    return [
+        line.rstrip()
+        for line in f
+        if not (line.isspace() or line.startswith('#'))
+    ]
+
+
 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=open(os.path.join(_CURRENT_DIR, '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']),
-    install_requires=[
-        'absl-py>=0.9.0',
-        'chex>=0.0.4',
-        'jax>=0.1.55',
-        'jaxlib>=0.1.37',
-        'dm-env>=1.2',
-        'numpy>=1.18.0',
-    ],
+    install_requires=_parse_requirements(
+        os.path.join(_CURRENT_DIR, 'requirements', 'requirements.txt')),
+    tests_require=_parse_requirements(
+        os.path.join(_CURRENT_DIR, 'requirements', 'requirements-test.txt')),
+    zip_safe=False,  # Required for full installation.
     python_requires='>=3.6',
     classifiers=[
-        'Development Status :: 3 - Alpha',
+        'Development Status :: 5 - Production/Stable',
         'Environment :: Console',
         'Intended Audience :: Science/Research',
         'License :: OSI Approved :: Apache Software License',