setup.py

#from glob import glob
from setuptools import find_packages, setup

# try:
#     from pybind11.setup_helpers import Pybind11Extension, build_ext
# except ImportError:
#     from setuptools import Extension as Pybind11Extension
# The above
# is not eeded any more because it is available at setup time due to pyproject.toml
# this is why setup_requires is also not needed

from pybind11.setup_helpers import Pybind11Extension, build_ext

import sys

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

exec(open('submodlib/version.py').read())

#ParallelCompile("NPY_NUM_BUILD_JOBS").install()

ext_modules = [
    Pybind11Extension(name="submodlib_cpp",
    sources=
        ["cpp/SetFunction.cpp",
        "cpp/utils/helper.cpp", "cpp/wrappers/wr_helper.cpp",
        "cpp/utils/sparse_utils.cpp", "cpp/wrappers/wr_sparse_utils.cpp",
        "cpp/optimizers/NaiveGreedyOptimizer.cpp", 
        "cpp/optimizers/LazyGreedyOptimizer.cpp", 
        "cpp/optimizers/StochasticGreedyOptimizer.cpp", 
        "cpp/optimizers/LazierThanLazyGreedyOptimizer.cpp", 
        "cpp/wrappers/wrapper.cpp", 
        "cpp/submod/FacilityLocation.cpp", "cpp/wrappers/wr_FacilityLocation.cpp", 
        "cpp/submod/FacilityLocation2.cpp", "cpp/wrappers/wr_FacilityLocation2.cpp", 
        "cpp/submod/DisparitySum.cpp", "cpp/wrappers/wr_DisparitySum.cpp", 
        "cpp/submod/FeatureBased.cpp", "cpp/wrappers/wr_FeatureBased.cpp", 
        "cpp/submod/GraphCut.cpp", "cpp/wrappers/wr_GraphCut.cpp", 
        "cpp/submod/SetCover.cpp", "cpp/wrappers/wr_SetCover.cpp", 
        "cpp/submod/ProbabilisticSetCover.cpp", "cpp/wrappers/wr_ProbabilisticSetCover.cpp", 
        "cpp/submod/DisparityMin.cpp", "cpp/wrappers/wr_DisparityMin.cpp", 
        "cpp/submod/LogDeterminant.cpp", "cpp/wrappers/wr_LogDeterminant.cpp", 
        "cpp/smi/FacilityLocationMutualInformation.cpp", "cpp/wrappers/wr_FacilityLocationMutualInformation.cpp", 
        "cpp/smi/FacilityLocationVariantMutualInformation.cpp", "cpp/wrappers/wr_FacilityLocationVariantMutualInformation.cpp", 
        "cpp/smi/ConcaveOverModular.cpp", "cpp/wrappers/wr_ConcaveOverModular.cpp", 
        "cpp/smi/GraphCutMutualInformation.cpp", "cpp/wrappers/wr_GraphCutMutualInformation.cpp", 
        "cpp/condgain/GraphCutConditionalGain.cpp", "cpp/wrappers/wr_GraphCutConditionalGain.cpp", 
        "cpp/condgain/ConditionalGain.cpp", 
        "cpp/condgain/FacilityLocationConditionalGain.cpp", "cpp/wrappers/wr_FacilityLocationConditionalGain.cpp", 
        "cpp/condgain/LogDeterminantConditionalGain.cpp", "cpp/wrappers/wr_LogDeterminantConditionalGain.cpp", 
        "cpp/condgain/ProbabilisticSetCoverConditionalGain.cpp", "cpp/wrappers/wr_ProbabilisticSetCoverConditionalGain.cpp", 
        "cpp/smi/ProbabilisticSetCoverMutualInformation.cpp", "cpp/wrappers/wr_ProbabilisticSetCoverMutualInformation.cpp", 
        "cpp/smi/MutualInformation.cpp", 
        "cpp/smi/LogDeterminantMutualInformation.cpp", "cpp/wrappers/wr_LogDeterminantMutualInformation.cpp", 
        "cpp/smi/SetCoverMutualInformation.cpp", "cpp/wrappers/wr_SetCoverMutualInformation.cpp", 
        "cpp/condgain/SetCoverConditionalGain.cpp", "cpp/wrappers/wr_SetCoverConditionalGain.cpp", 
        "cpp/cmi/FacilityLocationConditionalMutualInformation.cpp", "cpp/wrappers/wr_FacilityLocationConditionalMutualInformation.cpp", 
        "cpp/cmi/LogDeterminantConditionalMutualInformation.cpp", "cpp/wrappers/wr_LogDeterminantConditionalMutualInformation.cpp", 
        "cpp/cmi/SetCoverConditionalMutualInformation.cpp", "cpp/wrappers/wr_SetCoverConditionalMutualInformation.cpp", 
        "cpp/cmi/ProbabilisticSetCoverConditionalMutualInformation.cpp", "cpp/wrappers/wr_ProbabilisticSetCoverConditionalMutualInformation.cpp", 
        "cpp/Clustered.cpp", "cpp/wrappers/wr_Clustered.cpp"],
        extra_compile_args=['-O3'],
        #include_dirs=["cpp/"],
        # Example: passing in the version to the compiled code
        #sorted(glob("cpp/submod/*.cpp")),
        define_macros = [('VERSION_INFO', __version__)],
        ),
]

setup(
    name='submodlib',
    #packages=find_packages(include=['submodlib']),
    packages=['submodlib', 'submodlib/functions'],
    #packages=find_packages('submodlib'),
    #package_dir={'':'submodlib'},
    #version='0.0.2',
    version=__version__,
    description='submodlib is an efficient and scalable library for submodular optimization which finds its application in summarization, data subset selection, hyper parameter tuning etc.',
    long_description=long_description,
    long_description_content_type="text/markdown",
    author='Vishal Kaushal',
    cmdclass={"build_ext": build_ext},
    ext_modules=ext_modules,
    author_email='vishal.kaushal@gmail.com',
    url="https://github.com/vishkaush/submodlib",
    #url='http://pypi.python.org/pypi/submodlib/',
    #url="https://github.com/pypa/sampleproject",
    license='MIT',
    # install_requires=[
    #     "numpy >= 1.14.2",
    #     "scipy >= 1.0.0",
    #     "numba >= 0.43.0",
    #     "tqdm >= 4.24.0",
    #     "nose"
    # ],
    install_requires=["numpy>=1.22.0", "scipy", "scikit-learn", "numba"],
    #setup_requires=['pybind11','pytest-runner'],
    tests_require=['pytest'],
    #extras_require={"test": "pytest"},
    test_suite='tests',
    #classifiers=[
    #    "Programming Language :: Python :: 3",
    #    "License :: OSI Approved :: MIT License",
    #    "Operating System :: OS Independent",
    #],
    zip_safe=False 
)