forked from facebook/Ax
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
131 lines (111 loc) · 3.76 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
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
import os
from setuptools import find_packages, setup
# TODO: read pinned Botorch version from a shared source
PINNED_BOTORCH_VERSION = "0.8.0"
if os.environ.get("ALLOW_BOTORCH_LATEST"):
# allows a more recent previously installed version of botorch to remain
# if there is no previously installed version, installs the latest release
botorch_req = f"botorch>={PINNED_BOTORCH_VERSION}"
else:
botorch_req = f"botorch=={PINNED_BOTORCH_VERSION}"
REQUIRES = [
botorch_req,
"jinja2", # also a Plotly dep
"pandas",
"scipy",
"scikit-learn",
# ipywidgets 8.0.0 is incompatible with Plotly >=5.7.0
# https://github.com/plotly/plotly.py/issues/3686
"ipywidgets<8.0.0",
"plotly",
"typeguard",
]
# pytest-cov requires pytest >= 3.6
DEV_REQUIRES = [
"beautifulsoup4",
"black==22.3.0",
"flake8",
"hypothesis",
"Jinja2",
"pyfakefs==5.1.0",
"pytest>=4.6",
"pytest-cov",
"sphinx==5.3.0",
"sphinx-autodoc-typehints==1.19.5",
"torchvision>=0.5.0",
"nbconvert",
"jupyter-client==6.1.12",
"yappi",
]
MYSQL_REQUIRES = ["SQLAlchemy==1.4.17"]
NOTEBOOK_REQUIRES = ["jupyter"]
UNITTEST_MINIMAL_REQUIRES = [
"tensorboard", # For tensorboard unit tests.
"torchvision", # For torchvision unit tests.
"torchx", # For torchx unit tests.
"protobuf<4", # Temporary fix for tensorboard / ray import errors.
]
UNITTEST_REQUIRES = (
DEV_REQUIRES + MYSQL_REQUIRES + NOTEBOOK_REQUIRES + UNITTEST_MINIMAL_REQUIRES
)
TUTORIAL_REQUIRES = UNITTEST_REQUIRES + [
"psycopg2", # Used in example DBSettings in a tutorial (as part of postgres).
"ray", # Required for building RayTune tutorial notebook.
"tabulate", # Required for building RayTune tutorial notebook.
"tensorboardX", # Required for building RayTune tutorial notebook.
"matplotlib", # Required for building Multi-objective tutorial notebook.
"pyro-ppl", # Required for to call run_inference.
"pytorch-lightning", # For the early stopping tutorial.
]
def local_version(version):
"""
Patch in a version that can be uploaded to test PyPI
"""
return ""
def setup_package() -> None:
"""Used for installing the Ax package."""
with open("README.md", "r") as fh:
long_description = fh.read()
setup(
name="ax-platform",
description="Adaptive Experimentation",
author="Facebook, Inc.",
license="MIT",
url="https://github.com/facebook/Ax",
keywords=["Experimentation", "Optimization"],
classifiers=[
"Development Status :: 4 - Beta",
"Operating System :: POSIX :: Linux",
"Operating System :: MacOS :: MacOS X",
"Programming Language :: Python :: 3",
],
long_description=long_description,
long_description_content_type="text/markdown",
python_requires=">=3.8",
install_requires=REQUIRES,
packages=find_packages(),
package_data={
# include all js, css, and html files in the package
"": ["*.js", "*.css", "*.html"]
},
extras_require={
"dev": DEV_REQUIRES,
"mysql": MYSQL_REQUIRES,
"notebook": NOTEBOOK_REQUIRES,
"unittest": UNITTEST_REQUIRES,
"unittest_minimal": UNITTEST_MINIMAL_REQUIRES,
"tutorial": TUTORIAL_REQUIRES,
},
use_scm_version={
"write_to": "ax/version.py",
"local_scheme": local_version,
},
setup_requires=["setuptools_scm"],
)
if __name__ == "__main__":
setup_package()