From c988d71ddc2933ce8cc804dfbe35d6de7233b25e Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 14 Jun 2026 22:50:21 +0000
Subject: [PATCH] python3Packages.onnx-ir: fix safetensors 0.8.0 compat

---
 pkgs/development/python-modules/onnx-ir/default.nix | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pkgs/development/python-modules/onnx-ir/default.nix b/pkgs/development/python-modules/onnx-ir/default.nix
index 60c4a5425d0c..b243c78a1436 100644
--- a/pkgs/development/python-modules/onnx-ir/default.nix
+++ b/pkgs/development/python-modules/onnx-ir/default.nix
@@ -14,6 +14,8 @@
   sympy,
   typing-extensions,
 
+  fetchpatch,
+
   # tests
   onnxruntime,
   parameterized,
@@ -36,6 +38,17 @@ buildPythonPackage (finalAttrs: {
     hash = "sha256-vdo8BiE7m9Qr3JktgcPGDZfykjcf/VYY39tfhtzOrpA=";
   };
 
+  patches = [
+    # safetensors 0.8.0 changed `serialize_file` to require `TensorSpec` objects instead of plain
+    # dicts, breaking `save_safetensors`
+    # https://github.com/onnx/ir-py/pull/439
+    (fetchpatch {
+      name = "fix-safetensor-0.8.0-compat";
+      url = "https://github.com/onnx/ir-py/commit/c4780478d251f839ec1bacbea72d0a0948553285.patch";
+      hash = "sha256-99MeGU1B/534shCq9/TgrAPWscp+braXVcjAtnWQ3uY=";
+    })
+  ];
+
   build-system = [
     setuptools
   ];