48 строки
1.5 KiB
Python
48 строки
1.5 KiB
Python
|
from distutils.core import setup
|
||
|
|
from setuptools import find_packages
|
||
|
|
import os
|
||
|
|
|
||
|
|
this = os.path.dirname(__file__)
|
||
|
|
|
||
|
|
with open(os.path.join(this, "requirements.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.1"
|
||
|
|
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",
|
||
|
|
version=version_str,
|
||
|
|
description="Convert scikit-learn models to PyTorch",
|
||
|
|
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=[
|
||
|
|
"Development Status :: 1 - Beta",
|
||
|
|
"Environment :: Console",
|
||
|
|
"Intended Audience :: Developers",
|
||
|
|
"Programming Language :: Python",
|
||
|
|
"Programming Language :: Python :: 3.6",
|
||
|
|
"Programming Language :: Python :: 3.7",
|
||
|
|
"License :: OSI Approved :: MIT License",
|
||
|
|
],
|
||
|
|
)
|