diff options
author | GaƩtan Lepage <33058747+GaetanLepage@users.noreply.github.com> | 2024-06-26 21:53:15 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-06-26 21:53:15 +0200 |
commit | 4584edc9ac66d9cb77fa5e5a0b0c1fa2327aef7b (patch) | |
tree | e222afd174c7f0f8e84a5ef75b5b372c6916b795 /pkgs | |
parent | cb0b65ff84f3facddb79293077addeceb771fb16 (diff) | |
parent | 620cfcbbee064cdf1a48e44c55e16eb72913d527 (diff) |
Merge pull request #322609 from GaetanLepage/flax
python311Packages.flax: 0.8.4 -> 0.8.5
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/python-modules/flax/default.nix | 34 |
1 files changed, 21 insertions, 13 deletions
diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix index 7c443368bd41f..955177b9bbca3 100644 --- a/pkgs/development/python-modules/flax/default.nix +++ b/pkgs/development/python-modules/flax/default.nix @@ -1,32 +1,40 @@ { lib, buildPythonPackage, - cloudpickle, - einops, + pythonOlder, fetchFromGitHub, - jax, + + # build-system jaxlib, - keras, - matplotlib, + pythonRelaxDepsHook, + setuptools-scm, + + # dependencies + jax, msgpack, numpy, optax, orbax-checkpoint, - pytest-xdist, - pytestCheckHook, - pythonOlder, - pythonRelaxDepsHook, pyyaml, rich, - setuptools-scm, - tensorflow, tensorstore, typing-extensions, + + # checks + cloudpickle, + einops, + keras, + pytest-xdist, + pytestCheckHook, + tensorflow, + + # optional-dependencies + matplotlib, }: buildPythonPackage rec { pname = "flax"; - version = "0.8.4"; + version = "0.8.5"; pyproject = true; disabled = pythonOlder "3.9"; @@ -35,7 +43,7 @@ buildPythonPackage rec { owner = "google"; repo = "flax"; rev = "refs/tags/v${version}"; - hash = "sha256-ZwqKZdJ9LOfWTav5nE9xMsMw/DbryqQUuu5fqeugBzY="; + hash = "sha256-6WOFq0758gtNdrlWqSQBlKmWVIGe5e4PAaGrvHoGjr0="; }; build-system = [ |