Unverified Commit 5387b817 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python312Packages.jaxopt: disable failing test (#371423)

parents 93a0462d ed627141

pkgs/development/python-modules/jaxopt/default.nix

+19 −11
Original line number Diff line number Diff line
@@ -2,20 +2,24 @@ 
  lib,

  stdenv,

  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,

  fetchpatch,

  pytest-xdist,

  pytestCheckHook,



  # build-system

  setuptools,



  # dependencies

  absl-py,

  cvxpy,

  jax,

  jaxlib,

  matplotlib,

  numpy,

  optax,

  scipy,



  # tests

  cvxpy,

  optax,

  pytest-xdist,

  pytestCheckHook,

  scikit-learn,

}:
@@ -24,8 +28,6 @@ buildPythonPackage rec { 
  version = "0.8.3";

  pyproject = true;



  disabled = pythonOlder "3.8";



  src = fetchFromGitHub {

    owner = "google";

    repo = "jaxopt";
@@ -48,17 +50,16 @@ buildPythonPackage rec { 
  dependencies = [

    absl-py

    jax

    jaxlib

    matplotlib

    numpy

    scipy

  ];



  nativeCheckInputs = [

    pytest-xdist

    pytestCheckHook

    cvxpy

    optax

    pytest-xdist

    pytestCheckHook

    scikit-learn

  ];
@@ -74,6 +75,10 @@ buildPythonPackage rec { 
    [

      # https://github.com/google/jaxopt/issues/592

      "test_solve_sparse"



      # AssertionError: Not equal to tolerance rtol=1e-06, atol=1e-06

      # https://github.com/google/jaxopt/issues/618

      "test_binary_logit_log_likelihood"

    ]

    ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [

      # https://github.com/google/jaxopt/issues/577
@@ -81,6 +86,9 @@ buildPythonPackage rec { 
      "test_solve_sparse"

      "test_logreg_with_intercept_manual_loop3"



      # Flaky (AssertionError)

      "test_inv_hessian_product_pytree3"



      # https://github.com/google/jaxopt/issues/593

      # Makes the test suite crash

      "test_dtype_consistency"