tjh.dev
1{
2 lib,
3 stdenv,
4 buildPythonPackage,
5 fetchFromGitHub,
6
7 # build-system
8 setuptools,
9 setuptools-scm,
10
11 # dependencies
12 botorch,
13 ipywidgets,
14 jinja2,
15 markdown,
16 pandas,
17 plotly,
18 pyre-extensions,
19 scikit-learn,
20 scipy,
21 sympy,
22
23 # tests
24 pyfakefs,
25 pytestCheckHook,
26 sqlalchemy,
27 tabulate,
28}:
29
30buildPythonPackage rec {
31 pname = "ax-platform";
32 version = "1.0.0";
33 pyproject = true;
34
35 src = fetchFromGitHub {
36 owner = "facebook";
37 repo = "ax";
38 tag = version;
39 hash = "sha256-DFsV1w6J7bTZNUq9OYExDvfc7IfTcthGKAnRMNujRKI=";
40 };
41
42 env.ALLOW_BOTORCH_LATEST = "1";
43
44 build-system = [
45 setuptools
46 setuptools-scm
47 ];
48
49 dependencies = [
50 botorch
51 ipywidgets
52 jinja2
53 markdown
54 pandas
55 plotly
56 pyre-extensions
57 scikit-learn
58 scipy
59 sympy
60 ];
61
62 nativeCheckInputs = [
63 pyfakefs
64 # pytest-xdist
65 pytestCheckHook
66 sqlalchemy
67 tabulate
68 ];
69 pytestFlagsArray = [
70 # Hangs forever
71 "--deselect=ax/analysis/plotly/tests/test_top_surfaces.py::TestTopSurfacesAnalysis::test_online"
72 ];
73
74 disabledTestPaths = [
75 "ax/benchmark"
76 "ax/runners/tests/test_torchx.py"
77
78 # broken with sqlalchemy 2
79 "ax/core/tests/test_experiment.py"
80 "ax/service/tests/test_ax_client.py"
81 "ax/service/tests/test_scheduler.py"
82 "ax/service/tests/test_with_db_settings_base.py"
83 ];
84
85 disabledTests =
86 [
87 # sqlalchemy.exc.ArgumentError: Strings are not accepted for attribute names in loader options; please use class-bound attributes directly.
88 "SQAStoreUtilsTest"
89 "SQAStoreTest"
90
91 # ValueError: `db_settings` argument should be of type ax.storage.sqa_store
92 "test_get_next_trials_with_db"
93
94 # exact comparison of floating points
95 "test_optimize_l0_homotopy"
96 # AssertionError: 5 != 2
97 "test_get_standard_plots_moo"
98 # AssertionError: Expected 'warning' to be called once. Called 3 times
99 "test_validate_kwarg_typing"
100 # uses torch.equal
101 "test_convert_observations"
102 # broken with sqlalchemy 2
103 "test_sql_storage"
104 ]
105 ++ lib.optionals stdenv.hostPlatform.isDarwin [
106 # flaky on x86
107 "test_gen_with_expanded_parameter_space"
108 ];
109
110 pythonImportsCheck = [ "ax" ];
111
112 meta = {
113 description = "Platform for understanding, managing, deploying, and automating adaptive experiments";
114 homepage = "https://ax.dev/";
115 changelog = "https://github.com/facebook/Ax/releases/tag/${version}";
116 license = lib.licenses.mit;
117 maintainers = with lib.maintainers; [ veprbl ];
118 };
119}