diff options
| -rw-r--r-- | pkgs/development/python-modules/jmp/default.nix | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/pkgs/development/python-modules/jmp/default.nix b/pkgs/development/python-modules/jmp/default.nix index d45c684623724..2435d90accaf6 100644 --- a/pkgs/development/python-modules/jmp/default.nix +++ b/pkgs/development/python-modules/jmp/default.nix @@ -8,15 +8,13 @@ buildPythonPackage rec { pname = "jmp"; - # As of 2022-01-01, the latest stable version (0.0.2) fails tests with recent JAX versions, - # IIUC it's fixed in https://github.com/deepmind/jmp/commit/4969392f618d7733b265677143d8c81e44085867 - version = "unstable-2021-10-03"; + version = "0.0.4"; src = fetchFromGitHub { owner = "deepmind"; repo = pname; - rev = "260e5ba01f46b10c579a61393e6c7e546aeae93e"; - hash = "sha256-BTHy/jNf6LeV+x3GTI9MDBWLK6A5z2Z1TQyBkHMTeuE="; + rev = "refs/tags/v${version}"; + hash = "sha256-+PefZU1209vvf1SfF8DXiTvKYEnZ4y8iiIr8yKikx9Y="; }; # Wheel requires only `numpy`, but the import needs `jax`. |