cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

python311Packages.jaxopt: apply patch / disable failing tests after scipy 1.12 update

Browse source 
add patch and disable failing aarch64 linux tests
This commit is contained in:
annalee 2024-02-04 13:26:59 +00:00
parent b4781965cd
commit edcd5ae003
No known key found for this signature in database
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
pkgs/development/python-modules/jaxopt/default.nix
View file

@ -1,7 +1,9 @@
{ lib
, stdenv
, buildPythonPackage
, pythonOlder
, fetchFromGitHub
, fetchpatch
, pytest-xdist
, pytestCheckHook
, absl-py
@ -29,6 +31,16 @@ buildPythonPackage rec {
hash = "sha256-T/BHSnuk3IRuLkBj3Hvb/tFIb7Au25jjQtvwL28OU1U=";
};
patches = [
# fix failing tests from scipy 1.12 update
# https://github.com/google/jaxopt/pull/574
(fetchpatch {
name = "scipy-1.12-fix-tests.patch";
url = "https://github.com/google/jaxopt/commit/48b09dc4cc93b6bc7e6764ed5d333f9b57f3493b.patch";
hash = "sha256-v+617W7AhxA1Dzz+DBtljA4HHl89bRTuGi1QfatobNY=";
})
];
propagatedBuildInputs = [
absl-py
jax
@ -54,6 +66,13 @@ buildPythonPackage rec {
"jaxopt.tree_util"
];
disabledTests = lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
# https://github.com/google/jaxopt/issues/577
"test_binary_logit_log_likelihood"
"test_solve_sparse"
"test_logreg_with_intercept_manual_loop3"
];
meta = with lib; {
homepage = "https://jaxopt.github.io";
description = "Hardware accelerated, batchable and differentiable optimizers in JAX";