docling-ibm-models 1.2.1__tar.gz → 1.3.1__tar.gz

{docling_ibm_models-1.2.1 → docling_ibm_models-1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 1.2.1
+Version: 1.3.1
 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
@@ -19,12 +19,13 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: Pillow (>=10.0.0,<11.0.0)
+Requires-Dist: huggingface_hub (>=0.23,<1)
 Requires-Dist: jsonlines (>=3.1.0,<4.0.0)
 Requires-Dist: lxml (>=4.9.1,<5.0.0)
 Requires-Dist: mean_average_precision (>=2021.4.26.0,<2022.0.0.0)
 Requires-Dist: numpy (>=1.24.4,<2.0.0)
 Requires-Dist: onnxruntime (>=1.16.2,<2.0.0)
-Requires-Dist: opencv-python-headless (>=4.9.0.80,<5.0.0.0)
+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"
@@ -110,7 +111,7 @@ Below we list datasets used with their description, source, and ***"TableFormer
 
 ## Configuration file
 
-Example configuration can be seen inside test `tests/test_tf_predictor.py`
+Example configuration can be found inside test `tests/test_tf_predictor.py`
 These are the main sections of the configuration file:
 
 - `dataset`: The directory for prepared data and the parameters used during the data loading.
@@ -128,16 +129,13 @@ You can download the model weights and config files from the links:
 - [TableFormer Checkpoint](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/tableformer)
 - [beehive_v0.0.5](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/layout/beehive_v0.0.5)
 
-Place the downloaded files into `tests/test_data/model_artifacts/` directory.
-
 
 ## Inference Tests
 
-This contains unit tests for Docling models.
+You can run the inference tests for the models with:
 
-First download the model weights (see above), then run:
 ```
-./devtools/check_code.sh
+python -m pytest tests/
 ```
 
 This will also generate prediction and matching visualizations that can be found here:

{docling_ibm_models-1.2.1 → docling_ibm_models-1.3.1}/README.md

@@ -76,7 +76,7 @@ Below we list datasets used with their description, source, and ***"TableFormer
 
 ## Configuration file
 
-Example configuration can be seen inside test `tests/test_tf_predictor.py`
+Example configuration can be found inside test `tests/test_tf_predictor.py`
 These are the main sections of the configuration file:
 
 - `dataset`: The directory for prepared data and the parameters used during the data loading.
@@ -94,16 +94,13 @@ You can download the model weights and config files from the links:
 - [TableFormer Checkpoint](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/tableformer)
 - [beehive_v0.0.5](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/layout/beehive_v0.0.5)
 
-Place the downloaded files into `tests/test_data/model_artifacts/` directory.
-
 
 ## Inference Tests
 
-This contains unit tests for Docling models.
+You can run the inference tests for the models with:
 
-First download the model weights (see above), then run:
 ```
-./devtools/check_code.sh
+python -m pytest tests/
 ```
 
 This will also generate prediction and matching visualizations that can be found here:

{docling_ibm_models-1.2.1 → docling_ibm_models-1.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "1.2.1"  # DO NOT EDIT, updated automatically
+version = "1.3.1"  # 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"
@@ -37,7 +37,8 @@ jsonlines = "^3.1.0"
 Pillow = "^10.0.0"
 tqdm = "^4.64.0"
 mean_average_precision = "^2021.4.26.0"
-opencv-python-headless = { version = "^4.9.0.80" }
+opencv-python-headless = "^4.6.0.66"
+huggingface_hub = ">=0.23,<1"
 
 [tool.poetry.dev-dependencies]
 black = {extras = ["jupyter"], version = "^24.4.2"}