-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
69 lines (61 loc) · 2.46 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# --------------------------------------------------------------------------
import os
import re
from pathlib import Path
from setuptools import find_packages, setup
_cwd = Path(os.path.dirname(os.path.realpath(__file__)))
def _get_version(rel_path):
pattern = re.compile('__version__\s+=\s+[\"|\'](?P<version>.+)[\"\']')
abs_path = Path(_cwd / rel_path)
for line in abs_path.open():
match = pattern.match(line.strip())
if match:
return match['version']
raise RuntimeError("Unable to find version string.")
def _get_requirements(rel_path):
abs_path = Path(_cwd / rel_path)
return [_.strip() for _ in abs_path.open().readlines()]
CLASSIFIERS = [
"Development Status :: 2 - Pre-Alpha",
"Intended Audience :: Developers",
"License :: OSI Approved :: MIT License",
"Operating System :: POSIX :: Linux",
"Operating System :: Microsoft :: Windows",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development",
"Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules",
"Programming Language :: Python",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
]
LONG_DESCRIPTION = (
"Orchard is a collection of generic implementation of models that support both tensor & pipeline parallel"
" optimization. The ranked models can be generated using Olive (https://microsoft.github.io/Olive)."
)
DESCRIPTION = LONG_DESCRIPTION.split(".", maxsplit=1)[0] + "."
setup(
name="orchard",
version=_get_version("orchard/__init__.py"),
description=DESCRIPTION,
long_description=LONG_DESCRIPTION,
author="Microsoft Corporation",
author_email="[email protected]",
license="MIT License",
classifiers=CLASSIFIERS,
url="https://microsoft.github.io/Olive/", # FIXME!
download_url="https://github.com/devang-ml/orchard/tags",
packages=find_packages(include=["orchard*"]),
python_requires=">=3.8.0",
install_requires=_get_requirements("requirements.txt"),
extras_require={},
include_package_data=False,
package_data={},
data_files=[],
)