-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
114 lines (95 loc) · 4.08 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
"""
Usage:
Create ~/.pypirc with info:
[distutils]
index-servers =
pypi
[pypi]
repository: https://upload.pypi.org/legacy/
username: ...
password: ...
(dep: pip3 install --user twine setuptools)
(cleanup: rm dist/* _setup_info_generated.py)
python3 setup.py sdist
twine upload dist/*.tar.gz
See also MANIFEST.in for included files.
For debugging this script:
python3 setup.py sdist
pip3 install --user dist/*.tar.gz -v
(Without -v, all stdout/stderr from here will not be shown.)
"""
import os
import shutil
from i6_models.__setup__ import get_version_str, debug_print_file
def main():
"""
Setup main entry
"""
# Do not use current time as fallback for the version anymore,
# as this would result in a version which can be bigger than what we actually have,
# so this would not be useful at all.
long_version = get_version_str(verbose=True, fallback="0.0.1+setup-fallback-version", long=True)
version = long_version[: long_version.index("+")]
if os.environ.get("DEBUG", "") == "1":
debug_print_file(".")
debug_print_file("PKG-INFO")
debug_print_file("pip-egg-info")
debug_print_file("pip-egg-info/i6_models.egg-info")
debug_print_file("pip-egg-info/i6_models.egg-info/SOURCES.txt") # like MANIFEST
if os.path.exists("PKG-INFO"):
if os.path.exists("MANIFEST"):
print("package_data, found PKG-INFO and MANIFEST")
package_data = open("MANIFEST").read().splitlines() + ["PKG-INFO"]
else:
print("package_data, found PKG-INFO, no MANIFEST, use *")
# Currently the setup will ignore all other data except in i6_models/.
# At least make the version available.
shutil.copy("PKG-INFO", "i6_models/")
shutil.copy("_setup_info_generated.py", "i6_models/")
# Just using package_data = ["*"] would only take files from current dir.
package_data = []
for root, dirs, files in os.walk("."):
for file in files:
package_data.append(os.path.join(root, file))
else:
print("dummy package_data, does not matter, likely you are running sdist")
with open("_setup_info_generated.py", "w") as f:
f.write("version = %r\n" % version)
f.write("long_version = %r\n" % long_version)
package_data = ["MANIFEST", "_setup_info_generated.py"]
from setuptools import setup
authors = [
dict(name="Christoph Lüscher", email="[email protected]"),
dict(name="Nick Rossenbach", email="[email protected]"),
dict(name="Benedikt Hilmes", email="[email protected]"),
dict(name="Jingjing Xu", email="[email protected]"),
dict(name="Mohammad Zeineldeen", email="[email protected]"),
dict(name="Albert Zeyer", email="[email protected]"),
dict(name="Peter Vieting", email="[email protected]"),
dict(name="Simon Berger", email="[email protected]"),
dict(name="Eugen Beck", email="[email protected]"),
dict(name="Ping Zheng", email="[email protected]"),
dict(name="Wilfried Michel", email="[email protected]"),
]
setup(
name="i6_models",
version=version,
packages=["i6_models"],
include_package_data=True,
package_data={"i6_models": package_data}, # filtered via MANIFEST.in
description="A collection of PyTorch NN models and parts.",
author=", ".join(d["name"] for d in authors),
author_email=", ".join(d["email"] for d in authors),
url="https://github.com/rwth-i6/i6_models/",
license="Mozilla Public License 2.0",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
# https://pypi.org/classifiers/
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
"Operating System :: OS Independent",
],
)
if __name__ == "__main__":
main()