|
8 | 8 | from setuptools import setup, find_packages
|
9 | 9 |
|
10 | 10 |
|
11 |
| -def read_requirements(path): |
12 |
| - requirements = [] |
| 11 | +def read_requirements(path_): |
| 12 | + requirements_ = [] |
13 | 13 |
|
14 |
| - with pathlib.Path(path).open() as requirements_txt: |
| 14 | + with pathlib.Path(path_).open() as requirements_txt: |
15 | 15 | for line in requirements_txt:
|
16 | 16 | if line.startswith("git+"):
|
17 | 17 | pkg_name = re.search(r"egg=([a-zA-Z0-9_-]+)", line.strip()).group(1)
|
18 |
| - requirements.append(pkg_name + " @ " + line.strip()) |
| 18 | + requirements_.append(pkg_name + " @ " + line.strip()) |
19 | 19 | else:
|
20 |
| - requirements.append(line.strip()) |
| 20 | + requirements_.append(line.strip()) |
21 | 21 |
|
22 |
| - return requirements |
| 22 | + return requirements_ |
23 | 23 |
|
24 | 24 |
|
25 | 25 | requirements = read_requirements("requirements/prod.txt")
|
26 | 26 | extra_requirements_dev = read_requirements("requirements/dev.txt")
|
27 | 27 | extra_requirements_cubit = read_requirements("requirements/cubit.txt")
|
28 | 28 | extra_requirements_torch = read_requirements("requirements/torch.txt")
|
| 29 | +extra_requirements_cli = read_requirements("requirements/cli.txt") |
29 | 30 |
|
30 | 31 | here = path.abspath(path.dirname(__file__))
|
31 | 32 |
|
@@ -62,6 +63,7 @@ def read_requirements(path):
|
62 | 63 | extras_require={
|
63 | 64 | "dev": extra_requirements_dev + extra_requirements_torch,
|
64 | 65 | "torch": extra_requirements_torch,
|
| 66 | + "cli": extra_requirements_cli, |
65 | 67 | },
|
66 | 68 | classifiers=[
|
67 | 69 | "Development Status :: 3 - Alpha",
|
|
0 commit comments