81 lines
1.6 KiB
Nix
81 lines
1.6 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, botorch
|
|
, ipywidgets
|
|
, jinja2
|
|
, pandas
|
|
, plotly
|
|
, setuptools
|
|
, setuptools-scm
|
|
, typeguard
|
|
, wheel
|
|
, hypothesis
|
|
, mercurial
|
|
, pyfakefs
|
|
, pytestCheckHook
|
|
, yappi
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "ax";
|
|
version = "0.3.4";
|
|
format = "pyproject";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "facebook";
|
|
repo = pname;
|
|
rev = version;
|
|
hash = "sha256-Yc6alEKXbtQ0hitIdPhkJWhZQg150b0NJJRLZ+f1hdY=";
|
|
};
|
|
|
|
nativeBuildInputs = [
|
|
setuptools
|
|
setuptools-scm
|
|
wheel
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
botorch
|
|
ipywidgets
|
|
jinja2
|
|
pandas
|
|
plotly
|
|
typeguard
|
|
];
|
|
|
|
SETUPTOOLS_SCM_PRETEND_VERSION = version;
|
|
|
|
checkInputs = [
|
|
hypothesis
|
|
mercurial
|
|
pyfakefs
|
|
pytestCheckHook
|
|
yappi
|
|
];
|
|
pytestFlagsArray = [
|
|
"--ignore=ax/benchmark"
|
|
"--ignore=ax/runners/tests/test_torchx.py"
|
|
# requires pyre_extensions
|
|
"--ignore=ax/telemetry/tests"
|
|
"--ignore=ax/core/tests/test_utils.py"
|
|
"--ignore=ax/early_stopping/tests/test_strategies.py"
|
|
# broken with sqlalchemy 2
|
|
"--ignore=ax/service/tests/test_ax_client.py"
|
|
"--ignore=ax/service/tests/test_scheduler.py"
|
|
"--ignore=ax/service/tests/test_with_db_settings_base.py"
|
|
"--ignore=ax/storage"
|
|
];
|
|
disabledTests = [
|
|
# exact comparison of floating points
|
|
"test_optimize_l0_homotopy"
|
|
];
|
|
pythonImportsCheck = [ "ax" ];
|
|
|
|
meta = with lib; {
|
|
description = "Ax is an accessible, general-purpose platform for understanding, managing, deploying, and automating adaptive experiments";
|
|
homepage = "https://ax.dev/";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ veprbl ];
|
|
};
|
|
}
|