docling-ibm-models 1.0.0__tar.gz → 1.1.1__tar.gz

{docling_ibm_models-1.0.0 → docling_ibm_models-1.1.1}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 1.0.0
+Version: 1.1.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
 Author: Nikos Livathinos
 Author-email: nli@zurich.ibm.com
-Requires-Python: >=3.11,<4.0
+Requires-Python: >=3.10,<4.0
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -14,6 +14,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: MacOS :: MacOS X
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -27,13 +28,13 @@ Requires-Dist: numpy (>=1.24.4,<2.0.0)
 Requires-Dist: onnxruntime (>=1.16.2,<2.0.0)
 Requires-Dist: opencv-python (>=4.9.0.80,<5.0.0.0) ; sys_platform != "linux"
 Requires-Dist: opencv-python-headless (>=4.9.0.80,<5.0.0.0) ; sys_platform == "linux"
-Requires-Dist: torch (==2.2.2)
-Requires-Dist: torchvision (==0.17.2)
+Requires-Dist: torch (>=2.2.2,<3.0.0)
+Requires-Dist: torchvision (>=0.17.2)
 Requires-Dist: tqdm (>=4.64.0,<5.0.0)
 Description-Content-Type: text/markdown
 
 [![PyPI version](https://img.shields.io/pypi/v/docling-ibm-models)](https://pypi.org/project/docling-ibm-models/)
-![Python](https://img.shields.io/badge/python-3.11%20%7C%203.12-blue)
+![Python](https://img.shields.io/badge/python-3.10%20|%203.11%20|%203.12-blue)
 [![Poetry](https://img.shields.io/endpoint?url=https://python-poetry.org/badge/v0.json)](https://python-poetry.org/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)
@@ -70,7 +71,7 @@ RUN curl -sSL 'https://install.python-poetry.org' > install-poetry.py \
 To install and run the package, simply set up a poetry environment
 
 ```
-poetry env use $(which python3.11)
+poetry env use $(which python3.10)
 poetry shell
 ```
 

{docling_ibm_models-1.0.0 → docling_ibm_models-1.1.1}/README.md

@@ -1,5 +1,5 @@
 [![PyPI version](https://img.shields.io/pypi/v/docling-ibm-models)](https://pypi.org/project/docling-ibm-models/)
-![Python](https://img.shields.io/badge/python-3.11%20%7C%203.12-blue)
+![Python](https://img.shields.io/badge/python-3.10%20|%203.11%20|%203.12-blue)
 [![Poetry](https://img.shields.io/endpoint?url=https://python-poetry.org/badge/v0.json)](https://python-poetry.org/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)
@@ -36,7 +36,7 @@ RUN curl -sSL 'https://install.python-poetry.org' > install-poetry.py \
 To install and run the package, simply set up a poetry environment
 
 ```
-poetry env use $(which python3.11)
+poetry env use $(which python3.10)
 poetry shell
 ```
 

{docling_ibm_models-1.0.0 → docling_ibm_models-1.1.1}/docling_ibm_models/layoutmodel/layout_predictor.py

@@ -118,7 +118,7 @@ class LayoutPredictor:
         }
         return info
 
-    def predict(self, orig_img: Union[Image, np.array]) -> Iterable[dict]:
+    def predict(self, orig_img: Union[Image.Image, np.ndarray]) -> Iterable[dict]:
         r"""
         Predict bounding boxes for a given image.
         The origin (0, 0) is the top-left corner and the predicted bbox coords are provided as:

{docling_ibm_models-1.0.0 → docling_ibm_models-1.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "1.0.0"  # DO NOT EDIT, updated automatically
+version = "1.1.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"
@@ -21,9 +21,9 @@ packages = [
 ]
 
 [tool.poetry.dependencies]
-python = "^3.11"
-torch = "2.2.2"
-torchvision = "0.17.2"
+python = "^3.10"
+torch = "^2.2.2"
+torchvision = ">=0.17.2"
 onnxruntime = "^1.16.2"
 numpy = "^1.24.4"
 lxml = "^4.9.1"
@@ -57,13 +57,13 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.black]
 line-length = 88
-target-version = ["py311"]
+target-version = ["py310"]
 include = '\.pyi?$'
 
 [tool.isort]
 profile = "black"
 line_length = 88
-py_version=311
+py_version=310
 
 [tool.semantic_release]
 # for default values check: