setup.py

import pathlib

from setuptools import find_packages, setup

here = pathlib.Path(__file__).parent.resolve()

long_description = (here / 'README.md').read_text(encoding='utf-8')

install_requires = [
    'numpy >= 1.20.2', 'scipy >= 1.6.0', 'absl-py >= 0.12.0',
    'gym[mujoco] >= 0.18.0', 'gdown >= 3.12.2', 'tqdm >= 4.60.0',
    'flax >= 0.3.5', 'jax >= 0.2.21', 'ml_collections >= 0.1.0',
    'optax >= 0.0.6', 'tensorboardX == 2.1',
    'tensorflow-probability >= 0.13.0', 'imageio >= 2.9.0',
    'imageio-ffmpeg >= 0.4.3',
    'd4rl @ git+https://github.com/rail-berkeley/d4rl@master#egg=d4rl'
]

description = ('Implementations of Reinforcement Learning algorithms in Jax.')

setup(
    name='jaxrl',
    version='0.0.6',
    description=description,
    long_description=long_description,
    long_description_content_type='text/markdown',
    url='https://github.com/ikostrikov/jaxrl',
    author='Ilya Kostrikov',
    classifiers=[
        'Development Status :: 3 - Alpha',
        'Intended Audience :: Developers',
        'Intended Audience :: Education',
        'Intended Audience :: Science/Research',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
    ],
    keywords='reinforcement, machine, learning, research',
    packages=find_packages(),
    install_requires=install_requires,
    license='MIT',
)