ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-10-12 23:48:25 +03:00
Code Issues Projects Releases Wiki Activity

python311Packages.jax: 0.4.25 -> 0.4.28

Browse Source 
Changelog:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-28
This commit is contained in:
Gaetan Lepage 2024-04-04 09:22:58 +02:00
parent 1380cf0f0b
commit cdf5d71153
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkgs/development/python-modules/jax/default.nix
View File

@ -29,7 +29,7 @@ let
in
buildPythonPackage rec {
pname = "jax";
version = "0.4.25";
version = "0.4.28";
pyproject = true;
disabled = pythonOlder "3.9";
@ -39,7 +39,7 @@ buildPythonPackage rec {
repo = "jax";
# google/jax contains tags for jax and jaxlib. Only use jax tags!
rev = "refs/tags/jax-v${version}";
hash = "sha256-poQQo2ZgEhPYzK3aCs+BjaHTNZbezJAECd+HOdY1Yok=";
hash = "sha256-qSHPwi3is6Ts7pz5s4KzQHBMbcjGp+vAOsejW3o36Ek=";
};
nativeBuildInputs = [
@ -81,6 +81,14 @@ buildPythonPackage rec {
"tests/"
];
# Prevents `tests/export_back_compat_test.py::CompatTest::test_*` tests from failing on darwin with
# PermissionError: [Errno 13] Permission denied: '/tmp/back_compat_testdata/test_*.py'
# See https://github.com/google/jax/blob/jaxlib-v0.4.27/jax/_src/internal_test_util/export_back_compat_test_util.py#L240-L241
# NOTE: this doesn't seem to be an issue on linux
preCheck = lib.optionalString stdenv.isDarwin ''
export TEST_UNDECLARED_OUTPUTS_DIR=$(mktemp -d)
'';
disabledTests = [
# Exceeds tolerance when the machine is busy
"test_custom_linear_solve_aux"