setup.py

from os import path
import setuptools
from shutil import copyfile
from sys import platform
import os


from wheel.bdist_wheel import bdist_wheel as _bdist_wheel


class bdist_wheel(_bdist_wheel):
    def finalize_options(self):
        _bdist_wheel.finalize_options(self)
        self.root_is_pure = False


dirname = path.dirname(path.abspath(__file__))

if platform == "linux" or platform == "linux2":
    lib_path = path.abspath(path.join(dirname, "../build/lib/libthundersvm.so"))
elif platform == "win32":
    lib_path = path.abspath(path.join(dirname, "../build/bin/Debug/thundersvm.dll"))
elif platform == "darwin":
    lib_path = path.abspath(path.join(dirname, "../build/lib/libthundersvm.dylib"))
else:
    raise EnvironmentError("OS not supported!")

if not path.exists(path.join(dirname, "thundersvm", path.basename(lib_path))):
    copyfile(lib_path, path.join(dirname, "thundersvm", path.basename(lib_path)))

DEVICE = "+" + os.environ["DEVICE"] if "DEVICE" in os.environ else ""
setuptools.setup(
    name="thundersvm",
    version="0.3.4" + DEVICE,
    packages=["thundersvm"],
    package_dir={"python": "thundersvm"},
    description="A Fast SVM Library on GPUs and CPUs",
    long_description="The mission of ThunderSVM is to help users easily and efficiently apply SVMs to solve problems. ThunderSVM exploits GPUs and multi-core CPUs to achieve high efficiency",
    long_description_content_type="text/plain",
    url="https://github.com/mvdb-enspi/thundersvm",
    package_data={"thundersvm": [path.basename(lib_path)]},
    setup_requires=["wheel"],
    install_requires=["numpy", "scipy", "scikit-learn"],
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: Apache Software License",
    ],
    python_requires=">=3",
    cmdclass={"bdist_wheel": bdist_wheel},
)