-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
46 lines (40 loc) · 1.23 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
import pathlib
from setuptools import find_packages, setup
import os
here = pathlib.Path(__file__).parent.resolve()
# Get the long description from the README file
long_description = (here / "README.md").read_text(encoding="utf-8")
def read_requirements():
with open('requirements.txt', 'r') as file:
return [line.strip() for line in file if not line.startswith('#')]
dev_requires = [
"black>=22.1.0",
"importlib-resources>=5.4.0",
"ipython>=7.31.0",
"notebook>=7.0.4",
"pre-commit>=2.17.0",
"seaborn>=0.12.2",
]
test_requires = []
extras_require = { # Optional
"dev": dev_requires,
"test": test_requires,
}
setup_kwargs = {
"name": "sharcs",
"version": "0.1.0",
"description": "SHARCS: Routing to sub-networks with varying widths for sample adaptive inference efficiency.",
"long_description": long_description,
"author": "SHARCS team",
"author_email": None,
"maintainer": None,
"maintainer_email": None,
"url": None,
# "package_data": package_data,
"package_dir": {"": "src"},
"packages": find_packages("src"),
"install_requires": read_requirements(),
"extras_require": extras_require,
"python_requires": ">=3.8,<4.0",
}
setup(**setup_kwargs)