Unverified Commit 648f16ac authored by Martin Weinelt's avatar Martin Weinelt
Browse files

python312Packages.ax-platform: 0.4.3 -> 0.5.0 

https://github.com/facebook/Ax/releases/tag/0.5.0
parent 228d4def

pkgs/development/python-modules/ax-platform/default.nix

+30 −17
Original line number Diff line number Diff line
@@ -13,17 +13,19 @@ 
  pyfakefs,

  pyre-extensions,

  pytestCheckHook,

  pythonAtLeast,

  pythonOlder,

  setuptools-scm,

  setuptools,

  sqlalchemy,

  tabulate,

  typeguard,

  yappi,

}:



buildPythonPackage rec {

  pname = "ax-platform";

  version = "0.4.3";

  version = "0.5.0";

  pyproject = true;



  disabled = pythonOlder "3.10";
@@ -32,7 +34,7 @@ buildPythonPackage rec { 
    owner = "facebook";

    repo = "ax";

    tag = version;

    hash = "sha256-jmBjrtxqg4Iu3Qr0HRqjVfwURXzbJaGm+DBFNHYk/vA=";

    hash = "sha256-CMKdnPvzQ9tvU9/01mRaWi/Beuyo19CtaXNJCoiwLOw=";

  };



  env.ALLOW_BOTORCH_LATEST = "1";
@@ -62,16 +64,13 @@ buildPythonPackage rec { 
    mercurial

    pyfakefs

    pytestCheckHook

    tabulate

    yappi

  ] ++ lib.flatten (builtins.attrValues optional-dependencies);

  ] ++ lib.flatten (lib.attrValues optional-dependencies);



  disabledTestPaths = [

    "ax/benchmark"

    "ax/runners/tests/test_torchx.py"

    # requires pyre_extensions

    "ax/telemetry/tests"

    "ax/core/tests/test_utils.py"

    "ax/early_stopping/tests/test_strategies.py"

    # broken with sqlalchemy 2

    "ax/core/tests/test_experiment.py"

    "ax/service/tests/test_ax_client.py"
@@ -80,7 +79,8 @@ buildPythonPackage rec { 
    "ax/storage"

  ];



  disabledTests = [

  disabledTests =

    [

      # exact comparison of floating points

      "test_optimize_l0_homotopy"

      # AssertionError: 5 != 2
@@ -89,6 +89,19 @@ buildPythonPackage rec { 
      "test_validate_kwarg_typing"

      # uses torch.equal

      "test_convert_observations"

      # broken with sqlalchemy 2

      "test_sql_storage"

    ]

    ++ lib.optionals (pythonAtLeast "3.13") [

      #  Both `metric_aggregation` and `criterion` must be `ReductionCriterion`

      "test_SingleDiagnosticBestModelSelector_max_mean"

      "test_SingleDiagnosticBestModelSelector_min_mean"

      "test_SingleDiagnosticBestModelSelector_min_min"

      "test_SingleDiagnosticBestModelSelector_model_cv_kwargs"

      "test_init"

      "test_gen"

      # "use MIN or MAX" does not match "Both `metric_aggregation` and `criterion` must be `ReductionCriterion`

      "test_user_input_error"

    ];



  pythonImportsCheck = [ "ax" ];