setup.py (737B)
1 2 from setuptools import setup 3 from pybind11.setup_helpers import Pybind11Extension, build_ext 4 5 ext_modules = [ 6 Pybind11Extension( 7 "decision_tree", # Name of the Python module 8 sources=[ 9 "src/module.cpp", # Pybind11 wrapper source 10 "src/ELCClassifier.cpp", # Your custom class implementation 11 "src/TreeNode.cpp", # Dependency source 12 ], 13 include_dirs=["include"], # Directory for header files 14 extra_compile_args=["-std=c++11", ] # "-03"], # Enable C++11 standard 15 ), 16 ] 17 18 setup( 19 name="decision_tree", 20 version="0.1", 21 description="Pybind11 wrapper for ELCClassifier", 22 ext_modules=ext_modules, 23 cmdclass={"build_ext": build_ext}, 24 )