updated setup.py following the official one

matsui528 · Nov 2, 2023 · f2adda9 · f2adda9
1 parent 3d7be66
commit f2adda9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,5 +1,5 @@
+from pybind11.setup_helpers import Pybind11Extension, build_ext
 from setuptools import setup, Extension, find_packages
-from setuptools.command.build_ext import build_ext
 import sys
 import setuptools
 import re
@@ -19,7 +19,7 @@
 
 
 ext_modules = [
-    Extension(
+    Pybind11Extension(
         'main',
         ['src/main.cpp',
          'src/pqkmeans.cpp'],  # For c++ pqkmeans
@@ -115,7 +115,7 @@ def build_extensions(self):
     install_requires=requirements,
     setup_requires=['pybind11>=2.9'],
     ext_modules=ext_modules,
-    cmdclass={'build_ext': BuildExt},
+    cmdclass={'build_ext': build_ext},
     zip_safe=False,
     python_requires=">=3.6",
 )