diff --git a/pkgs/development/python-modules/docling-ibm-models/default.nix b/pkgs/development/python-modules/docling-ibm-models/default.nix
index 86c1c595d7b1..38137792b16c 100644
--- a/pkgs/development/python-modules/docling-ibm-models/default.nix
+++ b/pkgs/development/python-modules/docling-ibm-models/default.nix
@@ -2,31 +2,40 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
   poetry-core,
+
   # dependencies
-  torch,
-  torchvision,
-  transformers,
+  docling-core,
   huggingface-hub,
   jsonlines,
   numpy,
   opencv-python-headless,
   pillow,
-  tqdm,
+  pydantic,
   safetensors,
+  torch,
+  torchvision,
+  tqdm,
+  transformers,
+
+  # tests
+  datasets,
   pytestCheckHook,
+  writableTmpDirAsHomeHook,
 }:
 
 buildPythonPackage rec {
   pname = "docling-ibm-models";
-  version = "3.3.2";
+  version = "3.4.1";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "DS4SD";
     repo = "docling-ibm-models";
     tag = "v${version}";
-    hash = "sha256-8mqDgbTj5g6jXEumj16Me9NjHLCOdR+pXmAwn2dghfg=";
+    hash = "sha256-IgKP+4en0N6xLBjzwmddDk8z5P3J1vo+UMW+65a01Ww=";
   };
 
   build-system = [
@@ -34,21 +43,21 @@ buildPythonPackage rec {
   ];
 
   dependencies = [
+    docling-core
+    huggingface-hub
+    jsonlines
+    numpy
+    opencv-python-headless
+    pillow
+    pydantic
+    safetensors
     torch
     torchvision
-    transformers
-    numpy
-    jsonlines
-    pillow
     tqdm
-    opencv-python-headless
-    huggingface-hub
-    safetensors
+    transformers
   ];
 
   pythonRelaxDeps = [
-    "pillow"
-    "torchvision"
     "transformers"
     "numpy"
   ];
@@ -58,19 +67,18 @@ buildPythonPackage rec {
   ];
 
   nativeCheckInputs = [
+    datasets
     pytestCheckHook
+    writableTmpDirAsHomeHook
   ];
 
-  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
+    "test_layoutpredictor"
+    "test_readingorder"
+    "test_tf_predictor"
   ];
 
   meta = {