setup.py

# coding=utf-8
"""
The setup module.
"""
from __future__ import print_function, absolute_import

from setuptools import setup, find_packages
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel

from tensoralloy.extension.interp import build_ext as interp_build_ext


__version__ = "1.0"
__author__ = "Xin Chen"
__email__ = "Bismarrck@me.com"


extensions = [
    interp_build_ext.get_interp_extension(),
]


# noinspection PyMissingOrEmptyDocstring,PyPep8Naming,PyAttributeOutsideInit
class bdist_wheel(_bdist_wheel):

    def finalize_options(self):
        _bdist_wheel.finalize_options(self)
        # Mark us as not a pure python package
        self.root_is_pure = False


if __name__ == "__main__":

    packages = find_packages()
    if not packages:
        packages = ['tensoralloy']

    setup(
        name="tensoralloy",
        author=__author__,
        author_email=__email__,
        version=__version__,
        cmdclass={
            'bdist_wheel': bdist_wheel,
            "build_ext": interp_build_ext.BuildExtension
        },
        ext_modules=extensions,
        description="Tensor-graph based machine learning framework for alloys.",
        packages=packages,
        package_data={
            "": ["*.toml", "*.db", "*.cif"],
        },
        include_package_data=True,
        entry_points={
            'console_scripts': ['tensoralloy=tensoralloy.cli.entry:main']
        },
        python_requires=">=3.7.0",
    )