docling-ibm-models 2.0.1__tar.gz → 2.0.3__tar.gz

{docling_ibm_models-2.0.1 → docling_ibm_models-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 2.0.1
+Version: 2.0.3
 Summary: This package contains the AI models used by the Docling PDF conversion package
 License: MIT
 Keywords: docling,convert,document,pdf,layout model,segmentation,table structure,table former
@@ -26,10 +26,8 @@ Requires-Dist: mean_average_precision (>=2021.4.26.0,<2022.0.0.0)
 Requires-Dist: numpy (>=1.24.4,<2.0.0) ; python_version < "3.13"
 Requires-Dist: numpy (>=2.1.0,<3.0.0) ; python_version >= "3.13"
 Requires-Dist: opencv-python-headless (>=4.6.0.66,<5.0.0.0)
-Requires-Dist: torch (>=2.2.2,<2.3.0) ; sys_platform == "darwin" and platform_machine == "x86_64"
-Requires-Dist: torch (>=2.2.2,<3.0.0) ; sys_platform != "darwin" or platform_machine != "x86_64"
-Requires-Dist: torchvision (>=0,<1) ; sys_platform != "darwin" or platform_machine != "x86_64"
-Requires-Dist: torchvision (>=0.17.2,<0.18.0) ; sys_platform == "darwin" and platform_machine == "x86_64"
+Requires-Dist: torch (>=2.2.2,<3.0.0)
+Requires-Dist: torchvision (>=0,<1)
 Requires-Dist: tqdm (>=4.64.0,<5.0.0)
 Description-Content-Type: text/markdown
 
@@ -88,6 +86,14 @@ To update or add new dependencies from `pyproject.toml`, rebuild `poetry.lock`
 poetry update
 ```
 
+#### MacOS Intel
+
+When in development mode on MacOS with Intel chips, one can use compatible dependencies with
+
+```console
+poetry update --with mac_intel
+```
+
 
 ## Pipeline Overview
 ![Architecture](docs/tablemodel_overview_color.png)

{docling_ibm_models-2.0.1 → docling_ibm_models-2.0.3}/README.md

@@ -53,6 +53,14 @@ To update or add new dependencies from `pyproject.toml`, rebuild `poetry.lock`
 poetry update
 ```
 
+#### MacOS Intel
+
+When in development mode on MacOS with Intel chips, one can use compatible dependencies with
+
+```console
+poetry update --with mac_intel
+```
+
 
 ## Pipeline Overview
 ![Architecture](docs/tablemodel_overview_color.png)

{docling_ibm_models-2.0.1 → docling_ibm_models-2.0.3}/docling_ibm_models/layoutmodel/layout_predictor.py

@@ -153,11 +153,15 @@ class LayoutPredictor:
 
             # Check against threshold
             if score > self._threshold:
+                l = min(w, max(0, box[0]))
+                t = min(h, max(0, box[1]))
+                r = min(w, max(0, box[2]))
+                b = min(h, max(0, box[3]))
                 yield {
-                    "l": box[0],
-                    "t": box[1],
-                    "r": box[2],
-                    "b": box[3],
+                    "l": l,
+                    "t": t,
+                    "r": r,
+                    "b": b,
                     "label": label,
                     "confidence": score,
                 }

{docling_ibm_models-2.0.1 → docling_ibm_models-2.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "2.0.1"  # DO NOT EDIT, updated automatically
+version = "2.0.3"  # DO NOT EDIT, updated automatically
 description = "This package contains the AI models used by the Docling PDF conversion package"
 authors = ["Nikos Livathinos <nli@zurich.ibm.com>", "Maxim Lysak <mly@zurich.ibm.com>", "Ahmed Nassar <ahn@zurich.ibm.com>", "Christoph Auer <cau@zurich.ibm.com>", "Michele Dolfi <dol@zurich.ibm.com>", "Peter Staar <taa@zurich.ibm.com>"]
 license = "MIT"
@@ -22,14 +22,8 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-torch = [
-  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^2.2.2"},
-  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~2.2.2"}
-]
-torchvision = [
-  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^0"},
-  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~0.17.2"}
-]
+torch = "^2.2.2"
+torchvision = "^0"
 numpy = [
     { version = "^2.1.0", markers = 'python_version >= "3.13"' },
     { version = "^1.24.4", markers = 'python_version < "3.13"' },
@@ -42,7 +36,7 @@ mean_average_precision = "^2021.4.26.0"
 opencv-python-headless = "^4.6.0.66"
 huggingface_hub = ">=0.23,<1"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 black = {extras = ["jupyter"], version = "^24.4.2"}
 pytest = "^7.2.2"
 pre-commit = "^3.7.1"
@@ -56,10 +50,22 @@ pytest-flake8 = "^1.1.0"
 types-requests = "^2.31.0.2"
 flake8-pyproject = "^1.2.3"
 pylint = "^2.17.5"
-
-[tool.poetry.group.dev.dependencies]
 ipykernel = "^6.29.5"
 
+[tool.poetry.group.mac_intel]
+optional = true
+
+[tool.poetry.group.mac_intel.dependencies]
+torch = [
+  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^2.2.2"},
+  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~2.2.2"}
+]
+torchvision = [
+  {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^0"},
+  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~0.17.2"}
+]
+
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"