From 7932cca2f36101e2251af4a02b293dfd5b0956a4 Mon Sep 17 00:00:00 2001
From: happysalada <raphael@megzari.com>
Date: Sat, 1 Feb 2025 07:59:56 -0500
Subject: [PATCH] python3Packages.docling-ibm-models: 2.0.4 -> 3.3.0

docling-ibm-models: fix read onlyy file system

docling-ibm-models: disable broken tests
---
 .../docling-ibm-models/default.nix            | 34 ++++++++++++-------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/docling-ibm-models/default.nix b/pkgs/development/python-modules/docling-ibm-models/default.nix
index 7d0a39a5e9a8..bc1fdaae7846 100644
--- a/pkgs/development/python-modules/docling-ibm-models/default.nix
+++ b/pkgs/development/python-modules/docling-ibm-models/default.nix
@@ -3,28 +3,30 @@
   buildPythonPackage,
   fetchFromGitHub,
   poetry-core,
+  # dependencies
+  torch,
+  torchvision,
+  transformers,
   huggingface-hub,
   jsonlines,
-  mean-average-precision,
   numpy,
   opencv-python-headless,
   pillow,
-  torch,
-  torchvision,
   tqdm,
+  safetensors,
   pytestCheckHook,
 }:
 
 buildPythonPackage rec {
   pname = "docling-ibm-models";
-  version = "2.0.4";
+  version = "3.3.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "DS4SD";
     repo = "docling-ibm-models";
     tag = "v${version}";
-    hash = "sha256-QZvkkazxgkGuSQKIYI+YghH7pLlDSEbCGhg89gZsOpk=";
+    hash = "sha256-wxkHd+TCBibOTWO09JOsjX6oBtUxZ/9IOmyLdeptzeQ=";
   };
 
   build-system = [
@@ -32,21 +34,23 @@ buildPythonPackage rec {
   ];
 
   dependencies = [
-    huggingface-hub
-    jsonlines
-    mean-average-precision
-    numpy
-    opencv-python-headless
-    pillow
     torch
     torchvision
+    transformers
+    numpy
+    jsonlines
+    pillow
     tqdm
+    opencv-python-headless
+    huggingface-hub
+    safetensors
   ];
 
   pythonRelaxDeps = [
-    "mean_average_precision"
     "pillow"
     "torchvision"
+    "transformers"
+    "numpy"
   ];
 
   pythonImportsCheck = [
@@ -57,10 +61,16 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
+  preCheck = ''
+    export HOME="$TEMPDIR"
+  '';
+
   disabledTests = [
     # Requires network access
     "test_layoutpredictor"
     "test_tf_predictor"
+    "test_code_formula_predictor" # huggingface_hub.errors.LocalEntryNotFoundError
+    "test_figure_classifier" # huggingface_hub.errors.LocalEntryNotFoundError
   ];
 
   meta = {