Ax/pyproject.toml at main · Radical-AI/Ax · GitHub

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
[build-system]
requires = ["setuptools>=34.4", "wheel", "setuptools_scm"]
build-backend = "setuptools.build_meta"

[project]
name = "ax-platform"
description = "Adaptive Experimentation"
authors = [{name = "Meta Platforms, Inc."}]
license = "MIT"
license-files = ["LICENSE"]
readme = "README.md"
requires-python = ">=3.10"
keywords = ["Experimentation", "Optimization"]
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Operating System :: POSIX :: Linux",
    "Operating System :: MacOS :: MacOS X",
    "Programming Language :: Python :: 3",
]
dynamic = ["version"]

dependencies = [
    "botorch @ git+https://github.com/Radical-AI/botorch.git@a87a8ba97",
    "gpytorch @ git+https://github.com/cornellius-gp/gpytorch.git@8433c0b",
    "jinja2",  # also a Plotly dep
    "pandas",
    "scipy",
    "scikit-learn",
    "ipywidgets",
    # Needed for compatibility with ipywidgets >= 8.0.0
    "plotly>=5.12.0",
    "pyre-extensions",
    "sympy",
    "markdown",
]

[project.optional-dependencies]
dev = [
    "beautifulsoup4",
    "Jinja2",
    "pyfakefs",
    "pytest>=4.6",
    "pytest-cov",
    "sphinx",
    "sphinx-autodoc-typehints",
    "sphinx_rtd_theme",
    "lxml",
    "mdformat-myst",
]

mysql = [
    "SQLAlchemy==1.4.17",
]

notebook = [
    "jupyter",
]

unittest_minimal = [
    # For tensorboard unit tests (min req: numpy 2.0 compatibility).
    "tensorboard>=2.18.0",
    "torchx",  # For torchx unit tests.
    # Required for building RayTune tutorial notebook and
    # deserializing data for benchmark suites.
    "pyarrow",
    "xgboost>=3.0.2",  # For XGBoost surrogate models.
]

unittest = [
    "ax-platform[dev,mysql,notebook,unittest_minimal]",
]

tutorial = [
    "ax-platform[unittest]",
    "matplotlib",  # Required for building sklearn.ipynb tutorial notebook.
    "papermill",  # For executing the tutorials.
    "mdformat",
]

[project.urls]
Homepage = "https://ax.dev/"
Repository = "https://github.com/facebook/Ax"

[tool.setuptools.packages]
find = {}

[tool.setuptools.package-data]
"*" = ["*.js", "*.css", "*.html"]

[tool.setuptools_scm]
write_to = "ax/version.py"
local_scheme = "node-and-date"

[tool.usort]
first_party_detection = false

[tool.ufmt]
formatter = "ruff-api"