From fb4974c5ed6963da27cfe46a9cffc39847de0f67 Mon Sep 17 00:00:00 2001
From: wxt <3264117476@qq.com>
Date: Sun, 12 Jan 2025 19:49:56 +0800
Subject: [PATCH] python313Packages.flaxlib: fix build

---
 pkgs/development/python-modules/flaxlib/default.nix | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pkgs/development/python-modules/flaxlib/default.nix b/pkgs/development/python-modules/flaxlib/default.nix
index bb49634b60348..1a3a35ac30c54 100644
--- a/pkgs/development/python-modules/flaxlib/default.nix
+++ b/pkgs/development/python-modules/flaxlib/default.nix
@@ -49,6 +49,13 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
+  env = {
+    # https://github.com/google/flax/issues/4491
+    # Upstream should update Cargo.lock
+    # Enabling `PYO3_USE_ABI3_FORWARD_COMPATIBILITY` allows us to temporarily avoid the issue
+    PYO3_USE_ABI3_FORWARD_COMPATIBILITY = true;
+  };
+
   # This package does not have tests (yet ?)
   doCheck = false;