setup.py

"""Setup for the skift package."""

# !/usr/bin/env python
# -*- coding: utf-8 -*-
import os

import setuptools

INSTALL_REQUIRES = [
    'torch',
    'torchvision',
    'gym',
    'tensorboard',
    'tqdm'
]

INSTALL_REQUIRES_NOTBOOK = [
    'matplotlib', 'IPython', 'ipykernel'
]

DEV_REQUIRES = [
    # testing and coverage
    'pytest', 'coverage', 'pytest-cov', 'codecov', 'sphinx', 'sphinx-glpi-theme',
]

path = os.path.abspath(os.path.dirname(__file__))

with open(os.path.abspath(path + '/README.md')) as f:
    README = f.read()

setuptools.setup(
    author="french ai team",
    name='blobrl',
    version='0.1.3',
    license="Apache-2.0",
    description='Reinforcement learning with pytorch ',
    long_description=README,
    long_description_content_type='text/markdown',
    url='https://github.com/french-ai/reinforcement',
    packages=setuptools.find_packages(),
    include_package_data=True,
    download_url='https://github.com/french-ai/reinforcement/archive/V0.1.3.tar.gz',
    python_requires=">=3.6.1",
    install_requires=INSTALL_REQUIRES,
    extras_require={
        'dev': INSTALL_REQUIRES + INSTALL_REQUIRES_NOTBOOK + DEV_REQUIRES,
        "notebook": INSTALL_REQUIRES + INSTALL_REQUIRES_NOTBOOK
    },
    classifiers=[
        'Development Status :: 3 - Alpha',
        'License :: OSI Approved :: Apache Software License',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Topic :: Software Development :: Libraries',
        'Topic :: Software Development :: Libraries :: Python Modules',
        'Intended Audience :: Developers',
    ],
)