forked from wandb/wandb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
111 lines (100 loc) · 3.32 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
#!/usr/bin/env python
"""wandb setup."""
from setuptools import setup
with open("package_readme.md") as readme_file:
readme = readme_file.read()
with open("requirements.txt") as requirements_file:
requirements = requirements_file.read().splitlines()
with open("requirements_sweeps.txt") as sweeps_requirements_file:
sweeps_requirements = sweeps_requirements_file.read().splitlines()
test_requirements = ["mock>=2.0.0", "tox-pyenv>=1.0.3"]
gcp_requirements = ["google-cloud-storage"]
aws_requirements = ["boto3"]
azure_requirements = ["azure-storage-blob"]
grpc_requirements = ["grpcio>=1.27.2"]
service_requirements = []
kubeflow_requirements = ["kubernetes", "minio", "google-cloud-storage", "sh"]
media_requirements = [
"numpy",
"moviepy",
"pillow",
"bokeh",
"soundfile",
"plotly",
"rdkit-pypi",
]
launch_requirements = [
"nbconvert",
"nbformat",
"chardet",
"iso8601",
"typing_extensions",
"boto3",
"botocore",
"google-auth",
"google-cloud-compute",
"google-cloud-storage",
"google-cloud-artifact-registry",
"kubernetes",
]
models_requirements = ["cloudpickle"]
setup(
name="wandb",
version="0.13.11.dev1",
description="A CLI and library for interacting with the Weights and Biases API.",
long_description=readme,
long_description_content_type="text/markdown",
author="Weights & Biases",
author_email="[email protected]",
url="https://github.com/wandb/wandb",
packages=["wandb"],
package_dir={"wandb": "wandb"},
package_data={"wandb": ["py.typed"]},
entry_points={
"console_scripts": [
"wandb=wandb.cli.cli:cli",
"wb=wandb.cli.cli:cli",
]
},
include_package_data=True,
install_requires=requirements,
license="MIT license",
zip_safe=False,
# keywords='wandb',
python_requires=">=3.6",
classifiers=[
"Development Status :: 5 - Production/Stable",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Natural Language :: English",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries :: Python Modules",
"Topic :: System :: Logging",
"Topic :: System :: Monitoring",
],
test_suite="tests",
tests_require=test_requirements,
extras_require={
"kubeflow": kubeflow_requirements,
"gcp": gcp_requirements,
"aws": aws_requirements,
"azure": azure_requirements,
"service": service_requirements,
"grpc": grpc_requirements,
"media": media_requirements,
"sweeps": sweeps_requirements,
"launch": launch_requirements,
"models": models_requirements,
},
)
# if os.name == "nt" and sys.version_info >= (3, 6):
# legacy_env_var = "PYTHONLEGACYWINDOWSSTDIO"
# if legacy_env_var not in os.environ:
# if os.system("setx " + legacy_env_var + " 1") != 0:
# raise Exception("Error setting environment variable " + legacy_env_var)