From 379aeabd483dfdd093ed2ea9b30726032b102a35 Mon Sep 17 00:00:00 2001
From: "R. Ryantm" <ryantm-bot@ryantm.com>
Date: Sat, 30 Nov 2024 08:47:33 +0000
Subject: [PATCH] python312Packages.dm-haiku: 0.0.12 -> 0.0.13

---
 .../python-modules/dm-haiku/default.nix       | 23 ++++++++++++++-----
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/pkgs/development/python-modules/dm-haiku/default.nix b/pkgs/development/python-modules/dm-haiku/default.nix
index e5f44feb69c5..715aa1f8c3a4 100644
--- a/pkgs/development/python-modules/dm-haiku/default.nix
+++ b/pkgs/development/python-modules/dm-haiku/default.nix
@@ -3,8 +3,10 @@
   buildPythonPackage,
   fetchFromGitHub,
   fetchpatch,
+  setuptools,
   absl-py,
   flax,
+  jax,
   jaxlib,
   jmp,
   numpy,
@@ -25,14 +27,14 @@
 let
   dm-haiku = buildPythonPackage rec {
     pname = "dm-haiku";
-    version = "0.0.12";
-    format = "setuptools";
+    version = "0.0.13";
+    pyproject = true;
 
     src = fetchFromGitHub {
       owner = "deepmind";
       repo = "dm-haiku";
       rev = "refs/tags/v${version}";
-      hash = "sha256-aJRXlMq4CNMH3ZSTDP8MgnVltdSc8l5raw4//KccL48=";
+      hash = "sha256-RJpQ9BzlbQ4X31XoJFnsZASiaC9fP2AdyuTAGINhMxs=";
     };
 
     patches = [
@@ -44,15 +46,24 @@ let
       })
     ];
 
-    propagatedBuildInputs = [
+    build-system = [ setuptools ];
+
+    dependencies = [
       absl-py
-      flax
-      jaxlib
+      jaxlib # implicit runtime dependency
       jmp
       numpy
       tabulate
     ];
 
+    optional-dependencies = {
+      jax = [
+        jax
+        jaxlib
+      ];
+      flax = [ flax ];
+    };
+
     pythonImportsCheck = [ "haiku" ];
 
     nativeCheckInputs = [