python312Packages.dm-haiku: cleanup & disable on python 3.13 (#385066)

pkgs/development/python-modules/dm-haiku/default.nix

@@ -1,16 +1,25 @@
 {
   lib,
   buildPythonPackage,
+  pythonAtLeast,
   fetchFromGitHub,
   fetchpatch,
+
+  # build-system
   setuptools,
+
+  # dependencies
   absl-py,
-  flax,
-  jax,
   jaxlib,
   jmp,
   numpy,
   tabulate,
+
+  # optional-dependencies
+  jax,
+  flax,
+
+  # tests
   pytest-xdist,
   pytestCheckHook,
   bsuite,
@@ -30,6 +39,9 @@ let
     version = "0.0.13";
     pyproject = true;
 
+    # ImportError: `haiku.experimental.flax` features require `flax` to be installed.
+    disabled = pythonAtLeast "3.13";
+
     src = fetchFromGitHub {
       owner = "deepmind";
       repo = "dm-haiku";
@@ -117,11 +129,11 @@ let
       dontInstall = true;
     });
 
-    meta = with lib; {
+    meta = {
       description = "Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet";
       homepage = "https://github.com/deepmind/dm-haiku";
-      license = licenses.asl20;
-      maintainers = with maintainers; [ ndl ];
+      license = lib.licenses.asl20;
+      maintainers = with lib.maintainers; [ ndl ];
     };
   };
 in