import setuptools
from Cython.Build import cythonize

VERSION = '1.1.0'

with open("README.md", "r", encoding="utf8") as fh:
    long_description = fh.read()

setuptools.setup(
    name="douzero",
    version=VERSION,
    author="Daochen Zha",
    author_email="daochen.zha@tamu.edu",
    description="DouZero DouDizhu AI",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/kwai/DouZero",
    license='Apache License 2.0',
    keywords=["DouDizhu", "AI", "Reinforcment Learning", "RL", "Torch", "Poker"],
    packages=setuptools.find_packages(),
    install_requires=[
        'torch',
        'rlcard'
    ],
    ext_modules = cythonize(
        ["douzero/dmc/*.py", "douzero/env/*.py", "douzero/evaluation/*.py", "douzero/radam/*.py"], 
        exclude="**/__init__.py",
        language_level=3
    ),
    requires_python='>=3.6',
    classifiers=[
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.6",
        "License :: OSI Approved :: Apache Software License",
        "Operating System :: OS Independent",
    ],
)