python312Packages.ax-platform: add optional-dependencies

- switch to disabledTestPaths
This commit is contained in:
Fabian Affolter 2024-11-01 09:03:55 +01:00
parent 730ebac01f
commit 2d7517cf9b

View File

@ -1,21 +1,24 @@
{ {
lib, lib,
botorch,
buildPythonPackage, buildPythonPackage,
fetchFromGitHub, fetchFromGitHub,
botorch, hypothesis,
ipywidgets, ipywidgets,
jinja2, jinja2,
jupyter,
mercurial,
pandas, pandas,
plotly, plotly,
setuptools,
setuptools-scm,
typeguard,
hypothesis,
mercurial,
pyfakefs, pyfakefs,
pytestCheckHook,
yappi,
pyre-extensions, pyre-extensions,
pytestCheckHook,
pythonOlder,
setuptools-scm,
setuptools,
sqlalchemy,
typeguard,
yappi,
}: }:
buildPythonPackage rec { buildPythonPackage rec {
@ -23,6 +26,8 @@ buildPythonPackage rec {
version = "0.4.3"; version = "0.4.3";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.10";
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "facebook"; owner = "facebook";
repo = "ax"; repo = "ax";
@ -30,6 +35,8 @@ buildPythonPackage rec {
hash = "sha256-jmBjrtxqg4Iu3Qr0HRqjVfwURXzbJaGm+DBFNHYk/vA="; hash = "sha256-jmBjrtxqg4Iu3Qr0HRqjVfwURXzbJaGm+DBFNHYk/vA=";
}; };
env.ALLOW_BOTORCH_LATEST = "1";
build-system = [ build-system = [
setuptools setuptools
setuptools-scm setuptools-scm
@ -45,7 +52,10 @@ buildPythonPackage rec {
pyre-extensions pyre-extensions
]; ];
env.ALLOW_BOTORCH_LATEST = "1"; optional-dependencies = {
mysql = [ sqlalchemy ];
notebook = [ jupyter ];
};
nativeCheckInputs = [ nativeCheckInputs = [
hypothesis hypothesis
@ -53,21 +63,23 @@ buildPythonPackage rec {
pyfakefs pyfakefs
pytestCheckHook pytestCheckHook
yappi yappi
]; ] ++ lib.flatten (builtins.attrValues optional-dependencies);
pytestFlagsArray = [
"--ignore=ax/benchmark" disabledTestPaths = [
"--ignore=ax/runners/tests/test_torchx.py" "ax/benchmark"
"ax/runners/tests/test_torchx.py"
# requires pyre_extensions # requires pyre_extensions
"--ignore=ax/telemetry/tests" "ax/telemetry/tests"
"--ignore=ax/core/tests/test_utils.py" "ax/core/tests/test_utils.py"
"--ignore=ax/early_stopping/tests/test_strategies.py" "ax/early_stopping/tests/test_strategies.py"
# broken with sqlalchemy 2 # broken with sqlalchemy 2
"--ignore=ax/core/tests/test_experiment.py" "ax/core/tests/test_experiment.py"
"--ignore=ax/service/tests/test_ax_client.py" "ax/service/tests/test_ax_client.py"
"--ignore=ax/service/tests/test_scheduler.py" "ax/service/tests/test_scheduler.py"
"--ignore=ax/service/tests/test_with_db_settings_base.py" "ax/service/tests/test_with_db_settings_base.py"
"--ignore=ax/storage" "ax/storage"
]; ];
disabledTests = [ disabledTests = [
# exact comparison of floating points # exact comparison of floating points
"test_optimize_l0_homotopy" "test_optimize_l0_homotopy"
@ -78,12 +90,13 @@ buildPythonPackage rec {
# uses torch.equal # uses torch.equal
"test_convert_observations" "test_convert_observations"
]; ];
pythonImportsCheck = [ "ax" ]; pythonImportsCheck = [ "ax" ];
meta = { meta = {
changelog = "https://github.com/facebook/Ax/releases/tag/${version}"; description = "Platform for understanding, managing, deploying, and automating adaptive experiments";
description = "Ax is an accessible, general-purpose platform for understanding, managing, deploying, and automating adaptive experiments";
homepage = "https://ax.dev/"; homepage = "https://ax.dev/";
changelog = "https://github.com/facebook/Ax/releases/tag/${version}";
license = lib.licenses.mit; license = lib.licenses.mit;
maintainers = with lib.maintainers; [ veprbl ]; maintainers = with lib.maintainers; [ veprbl ];
}; };