diff --git a/.gitignore b/.gitignore index d16c110..39070f9 100644 --- a/.gitignore +++ b/.gitignore @@ -6,3 +6,6 @@ douzero_checkpoints/ venv/ .idea/ *.egg-info/ +*.c +*.pyd +build/ diff --git a/setup.py b/setup.py index 554c30c..e0a5fa1 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,5 @@ import setuptools +from Cython.Build import cythonize VERSION = '1.1.0' @@ -21,6 +22,11 @@ setuptools.setup( '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",