nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-07-19 15:11:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.flax: fix test with new jax

Browse Source
This commit is contained in:
Michael Daniels
2026-06-19 22:06:11 -04:00
committed by Gaetan Lepage
parent defc2a7926
commit 183181afcc
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkgs/development/python-modules/flax/default.nix
View File

@@ -46,6 +46,14 @@ buildPythonPackage (finalAttrs: {
hash = "sha256-a78KiTsCCARWZvbxz9QKdUKnjkDJGXcPVVJu5rU4m/U=";
};
# DeprecationWarning: `with mesh:` context manager has been deprecated. Please use `with jax.set_mesh(mesh):` instead.
postPatch = ''
substituteInPlace tests/nnx/transforms_test.py \
--replace-fail \
"with mesh:" \
"with jax.set_mesh(mesh):"
'';
build-system = [
setuptools
setuptools-scm