from distutils.core import setup
from setuptools import find_packages
import os
this = os.path.dirname(__file__)
with open(os.path.join(this, "requirements", "common.txt"), "r") as f:
requirements = [_ for _ in [_.strip("\r\n ") for _ in f.readlines()] if _ is not None]
packages = find_packages()
assert packages
# read version from the package file.
version_str = "0.0.2"
with (open(os.path.join(this, "hummingbird/__init__.py"), "r")) as f:
line = [_ for _ in [_.strip("\r\n ") for _ in f.readlines()] if _.startswith("__version__")]
if len(line) > 0:
version_str = line[0].split("=")[1].strip('" ')
README = os.path.join(os.getcwd(), "README.md")
with open(README) as f:
long_description = f.read()
start_pos = long_description.find("## Introduction")
if start_pos >= 0:
long_description = long_description[start_pos:]
setup(
name="hummingbird-ml",
version=version_str,
description="Convert trained traditional machine learning models into tensor computations",
license="MIT License",
author="Microsoft Corporation",
author_email="hummingbird-dev@microsoft.com",
url="https://github.com/microsoft/hummingbird",
packages=packages,
include_package_data=True,
install_requires=requirements,
classifiers=[
"Environment :: Console",
"Intended Audience :: Developers",
"Programming Language :: Python",
"Operating System :: OS Independent",
"License :: OSI Approved :: MIT License",
],
python_requires=">=3.5",
)
