Migrate config from setup.py to pyproject.toml

Author: ymd-h

pyproject.toml

@@ -1,7 +1,44 @@
 [build-system]
-requires = ["setuptools", "wheel"]
+requires = ["setuptools", "wheel", "numpy>=2.0.0", "cython>=3.0.11"]
 build-backend = "setuptools.build_meta"
 
+[project]
+name = "cpprb"
+version = "11.0.0"
+license = { file = "LICENSE" }
+authors = [{ name = "Yamada Hiroyuki" }]
+classifiers=[
+  "Programming Language :: Python",
+  "Programming Language :: Python :: 3",
+  "License :: OSI Approved :: MIT License",
+  "Operating System :: OS Independent",
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "Intended Audience :: Science/Research",
+  "Topic :: Scientific/Engineering",
+  "Topic :: Scientific/Engineering :: Artificial Intelligence",
+  "Topic :: Software Development :: Libraries",
+]
+dependencies = ["numpy"]
+dynamic = [
+  "description",
+  "readme",
+]
+
+[project.urls]
+"Home Page" = "https://ymd_h.gitlab.io/cpprb/"
+"Source Code" = "https://gitlab.com/ymd_h/cpprb"
+"Mirror" = "https://github.com/ymd-h/cpprb"
+"Change Log" = "https://ymd_h.gitlab.io/cpprb/changelog/"
+"Bug Report & QA" = "https://github.com/ymd-h/cpprb/discussions"
+
+
+[tool.setuptools]
+packages = ["cpprb"]
+package-dir = { "" = "src" }
+include-package-data = false
+
+
 [tool.cibuildwheel]
 build = ["cp3{9,10,11,12,13}-{macosx_*,{win,????linux}_*64}"]
 build-frontend = "build[uv]"

setup.py

@@ -5,7 +5,7 @@
 import warnings
 
 from setuptools import Extension, setup
-from setuptools.command.build_ext import build_ext
+import numpy as np
 
 debug = os.getenv("DEBUG_CPPRB")
 
@@ -18,31 +18,6 @@
     or os.getenv("GITLAB_CI")
 )
 
-arm_mac = (sysconfig.get_platform().split("-")[0] == "macosx") and (platform.machine() == "arm64")
-
-requires = ["numpy"]
-setup_requires = ["wheel"]
-
-if sys.version_info >= (3, 9):
-    # NumPy 2.0 breaks ABI compatibility.
-    # To support both NumPy 1.x and 2.x, build with NumPy 2.x
-    # cf. https://numpy.org/devdocs/dev/depending_on_numpy.html#numpy-2-abi-handling
-    setup_requires.append("numpy>=2.0.0")
-elif (sys.version_info < (3, 9)) and not arm_mac:
-    # NumPy 1.20 breaks ABI compatibility.
-    # To support both NumPy 1.19.x and 1.20+, build with NumPy 1.19
-    # This is necessary to work with old library which requires old NumPy.
-
-    # NumPy 1.19.5 doesn't support Python 3.10+
-    # NumPy 1.19.5 doesn't distribute wheel for macOS aarch64,
-    setup_requires.append("numpy<1.20")
-else:
-    setup_requires.append("numpy")
-
-rb_source = "src/cpprb/PyReplayBuffer"
-cpp_ext = ".cpp"
-pyx_ext = ".pyx"
-
 
 # Set compiler flags depending on platform
 if platform.system() == "Windows":
@@ -61,59 +36,31 @@
     if debug:
         extra_compile_args.append("-DCYTHON_TRACE_NOGIL=1")
 
-# Check cythonize or not
-pyx_file = rb_source + pyx_ext
-use_cython = os.path.exists(pyx_file)
 
-if use_cython:
-    suffix = pyx_ext
-    setup_requires.extend(["cython>=3.0.11"])
-    compiler_directives = {"language_level": "3"}
 
-    if debug:
-        compiler_directives["linetrace"] = True
-else:
-    suffix = cpp_ext
+compiler_directives = {"language_level": "3"}
+if debug:
+    compiler_directives["linetrace"] = True
+
 
 # Set ext_module
 ext = [["cpprb", "PyReplayBuffer"], ["cpprb", "VectorWrapper"]]
 
 ext_modules = [
     Extension(
         ".".join(e),
-        sources=["src/" + "/".join(e) + suffix],
-        extra_compile_args=extra_compile_args,
-        extra_link_args=extra_link_args,
-        language="c++",
+        sources = ["src/" + "/".join(e) + ".pyx"],
+        include_dirs = [np.get_include()],
+        extra_compile_args = extra_compile_args,
+        extra_link_args = extra_link_args,
+        language = "c++",
+        compiler_directives = compiler_directives,
+        cython_include_dirs = ["."],
     )
     for e in ext
 ]
 
 
-class LazyImportBuildExtCommand(build_ext):
-    """
-    build_ext command class for lazy numpy and cython import
-    """
-
-    def run(self):
-        import numpy as np
-
-        self.include_dirs.append(np.get_include())
-        build_ext.run(self)
-
-    def finalize_options(self):
-        if use_cython:
-            from Cython.Build import cythonize
-
-            self.distribution.ext_modules = cythonize(
-                self.distribution.ext_modules,
-                compiler_directives=compiler_directives,
-                include_path=["."],
-                annotate=True,
-            )
-        super().finalize_options()
-
-
 description = "ReplayBuffer for Reinforcement Learning written by C++ and Cython"
 README = os.path.join(os.path.abspath(os.path.dirname(__file__)), "README.md")
 if os.path.exists(README):
@@ -126,36 +73,8 @@ def finalize_options(self):
     long_description_content_type = "text/plain"
 
 setup(
-    name="cpprb",
-    author="Yamada Hiroyuki",
     description=description,
-    version="11.0.0",
-    install_requires=requires,
-    setup_requires=setup_requires,
-    cmdclass={"build_ext": LazyImportBuildExtCommand},
-    url="https://ymd_h.gitlab.io/cpprb/",
-    project_urls={
-        "Source Code": "https://gitlab.com/ymd_h/cpprb",
-        "Mirror": "https://github.com/ymd-h/cpprb",
-        "Change Log": "https://ymd_h.gitlab.io/cpprb/changelog/",
-        "Bug Report & QA": "https://github.com/ymd-h/cpprb/discussions",
-    },
-    package_dir={"": "src"},
     ext_modules=ext_modules,
-    include_dirs=["cpprb"],
-    packages=["cpprb"],
-    classifiers=[
-        "Programming Language :: Python",
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-        "Development Status :: 4 - Beta",
-        "Intended Audience :: Developers",
-        "Intended Audience :: Science/Research",
-        "Topic :: Scientific/Engineering",
-        "Topic :: Scientific/Engineering :: Artificial Intelligence",
-        "Topic :: Software Development :: Libraries",
-    ],
     long_description=long_description,
     long_description_content_type=long_description_content_type,
 )