python3Packages.docling-ibm-models: 2.0.4 -> 3.3.0

docling-ibm-models: fix read onlyy file system

docling-ibm-models: disable broken tests

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 = {