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)))

build_tag = os.environ.get('BUILD_TAG', '')
setuptools.setup(name="thundersvm" + build_tag,
                 version="0.3.4",
                 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/Xtra-Computing/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},
                 )