From ff94bb4ba64bfc3c6efe05cdc303730b666477e5 Mon Sep 17 00:00:00 2001
From: natsukium <tomoya.otabi@gmail.com>
Date: Tue, 15 Aug 2023 00:23:43 +0900
Subject: [PATCH] python310Packages.jmp: unstable-2021-10-03 -> 0.0.4

Diff: https://github.com/deepmind/jmp/compare/4b94370b8de29b79d6f840b09d1990b91c1afddd...v0.0.4

Changelog: https://github.com/deepmind/jmp/releases/tag/v0.0.4
---
 pkgs/development/python-modules/jmp/default.nix | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/jmp/default.nix b/pkgs/development/python-modules/jmp/default.nix
index d45c68462372..2435d90accaf 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`.