setup.py

# Copyright 2021 Alibaba Group Holding Limited. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# =============================================================================

r'''HybridBackend setup script.
'''

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import os
from setuptools import Extension
from setuptools import find_packages
from setuptools import setup
from setuptools.command.build_ext import build_ext
from setuptools.dist import Distribution

from hybridbackend import __version__
from hybridbackend import __author__

NAME = 'hybridbackend{}'.format(os.getenv('WHEEL_ALIAS', ''))
VERSION = '{}{}'.format(__version__, os.getenv('WHEEL_BUILD', ''))
PACKAGES = find_packages(exclude=['cpp', 'tests', 'examples'])
PACKAGE_DATA = {'': ['*.so', '*.so.*']}
REQUIRES = os.getenv('WHEEL_REQUIRES', '').split(';')


class BinaryDistribution(Distribution):
  r'''This class is needed in order to create OS specific wheels.
  '''
  def has_ext_modules(self):
    return True

  def is_pure(self):
    return False


class NoExtensionBuilder(build_ext):
  r'''Build extensions to do nothing.
  '''
  def build_extension(self, ext):
    return


setup(
    name=NAME,
    version=VERSION,
    packages=PACKAGES,
    include_package_data=True,
    package_data=PACKAGE_DATA,
    install_requires=REQUIRES,
    ext_modules=[Extension('', sources=[])],
    cmdclass={'build_ext': NoExtensionBuilder},
    distclass=BinaryDistribution,
    zip_safe=False,
    author=__author__,
    description='Efficient training of deep recommenders on cloud.',
    long_description=(
        "HybridBackend is a training framework for deep recommenders which "
        "bridges gap between evolving cloud infrastructure and complex "
        "training process."),
    long_description_content_type='text/markdown',
    url="https://github.com/alibaba/HybridBackend",
    download_url='https://github.com/alibaba/HybridBackend/tags',
    project_urls={
        'Bug Tracker': 'https://github.com/alibaba/HybridBackend/issues',
        'Documentation': 'https://hybridbackend.readthedocs.io/en/latest/',
        'Source Code': 'https://github.com/alibaba/HybridBackend',
    },
    keywords=('deep learning', 'recommendation system'),
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: Apache Software License',
        'Programming Language :: Python :: 3.6',
        'Operating System :: POSIX :: Linux',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'Topic :: Software Development :: Libraries :: Python Modules',
        'Topic :: Software Development :: Libraries',
    ],
    license='Apache License 2.0',
    license_files=('LICENSE', 'NOTICE'),
)