From eea397c5920f2e4228b88e5d86ae689a351f3ef0 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 22 Apr 2026 08:45:52 +0000
Subject: [PATCH] python3Packages.flax: 0.12.6 -> 0.12.7

Diff: https://github.com/google/flax/compare/v0.12.6...v0.12.7

Changelog: https://github.com/google/flax/releases/tag/v0.12.7
---
 pkgs/development/python-modules/flax/default.nix | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix
index 119b841cde81..89f24260f486 100644
--- a/pkgs/development/python-modules/flax/default.nix
+++ b/pkgs/development/python-modules/flax/default.nix
@@ -13,7 +13,6 @@
   numpy,
   optax,
   orbax-checkpoint,
-  orbax-export,
   pyyaml,
   rich,
   tensorstore,
@@ -36,14 +35,15 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "flax";
-  version = "0.12.6";
+  version = "0.12.7";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "google";
     repo = "flax";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-rIDfF9W8cxF0njH4e4uhqURQ0C4N8Boe76u6meMgC34=";
+    hash = "sha256-a78KiTsCCARWZvbxz9QKdUKnjkDJGXcPVVJu5rU4m/U=";
   };
 
   build-system = [
@@ -58,7 +58,6 @@ buildPythonPackage (finalAttrs: {
     numpy
     optax
     orbax-checkpoint
-    orbax-export
     pyyaml
     rich
     tensorstore
@@ -78,11 +77,6 @@ buildPythonPackage (finalAttrs: {
     tensorflow
   ];
 
-  pytestFlags = [
-    # FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
-    "-Wignore::FutureWarning"
-  ];
-
   disabledTestPaths = [
     # Docs test, needs extra deps + we're not interested in it.
     "docs/_ext/codediff_test.py"