-
Notifications
You must be signed in to change notification settings - Fork 747
/
setup.py
145 lines (114 loc) · 4.53 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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
import distutils.cmd
import sys
from pathlib import Path
from textwrap import dedent
from setuptools import setup
ROOT = Path(__file__).parent
# Note: In GluonTS we use git tags to manage versions. A new release is created
# by creating a new tag on GitHub through their release mechanism. Thus,
# `gluonts.__version__` uses the latest available git tag. If there are
# additional commits on top of the tagged commit, we extend the version
# information and append a `.dev0+g{commit_id}` to the version. If there are
# uncommitted changes, an additional `.dirty` is appended to the version.
# Since we always rely on the latest available tag, it is important to ensure
# that the latest tag in the `dev` branch is `v0` and not a more specific
# version like `v0.x`, since the `dev` branch should be independent from a
# more specific version. This means that we can't tag a commit on `dev` when
# doing a new release. If git is not available, we fallback to version
# `0.0.0`. When doing releases, the version gets frozen, by overwriting
# `meta/_version.py` with the static version information. For this to work, we
# need to adapt the `sdist` and `build_py` command classes to also handle
# freezing of the versions.
def get_version_cmdclass(version_file) -> dict:
with open(version_file) as fobj:
code = fobj.read()
globals_ = {"__file__": str(version_file)}
exec(code, globals_)
# When `_version.py` is replaced, it should still contain `__version__`,
# but no longer "cmdclass".
if "cmdclass" not in globals_:
assert "__version__" in globals_
return {}
return globals_["cmdclass"]()
class TypeCheckCommand(distutils.cmd.Command):
"""
A custom command to run MyPy on the project sources.
"""
description = "run MyPy on Python source files"
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
# import here (after the setup_requires list is loaded),
# otherwise a module-not-found error is thrown
import mypy.api
excluded_folders = [
str(p.parent.relative_to(ROOT)) for p in ROOT.glob("src/**/.typeunsafe")
]
if len(excluded_folders) > 0:
print(
"The following folders contain a `.typeunsafe` marker file "
"and will *not* be type-checked with `mypy`:"
)
for folder in excluded_folders:
print(f" {folder}")
args = [str(ROOT / "src")]
for folder in excluded_folders:
args.append("--exclude")
args.append(folder)
std_out, std_err, exit_code = mypy.api.run(args)
print(std_out, file=sys.stdout)
print(std_err, file=sys.stderr)
if exit_code:
error_msg = dedent(
f"""
Mypy command
mypy {" ".join(args)}
returned a non-zero exit code. Fix the type errors listed above
and then run
python setup.py type_check
in order to validate your fixes.
"""
).lstrip()
print(error_msg, file=sys.stderr)
sys.exit(exit_code)
def find_requirements(filename):
with open(ROOT / "requirements" / filename) as fobj:
return [line.rstrip() for line in fobj if not line.startswith("#")]
arrow_require = find_requirements("requirements-arrow.txt")
docs_require = find_requirements("requirements-docs.txt")
tests_require = find_requirements("requirements-test.txt")
sagemaker_api_require = find_requirements(
"requirements-extras-sagemaker-sdk.txt"
)
shell_require = find_requirements("requirements-extras-shell.txt")
mxnet_require = find_requirements("requirements-mxnet.txt")
torch_require = find_requirements("requirements-pytorch.txt")
dev_require = (
arrow_require
+ docs_require
+ tests_require
+ shell_require
+ sagemaker_api_require
)
setup(
install_requires=find_requirements("requirements.txt"),
tests_require=tests_require,
extras_require={
"arrow": arrow_require,
"dev": dev_require,
"docs": docs_require,
"mxnet": mxnet_require,
"R": find_requirements("requirements-extras-r.txt"),
"Prophet": find_requirements("requirements-extras-prophet.txt"),
"pro": arrow_require + ["orjson"],
"shell": shell_require,
"torch": torch_require,
},
cmdclass={
"type_check": TypeCheckCommand,
**get_version_cmdclass("src/gluonts/meta/_version.py"),
},
)