From 0cb75206487ef4ec83d5baf90f94239f50865522 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Sun, 1 Jun 2025 09:06:41 +0200 Subject: [PATCH] restore stable imports Signed-off-by: Michele Dolfi --- docling/datamodel/document.py | 22 ++++++++++++++++++++++ tests/test_backend_html.py | 3 +-- tests/test_backend_msword.py | 4 +--- 3 files changed, 24 insertions(+), 5 deletions(-) diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py index acb33f1d..e464ee3b 100644 --- a/docling/datamodel/document.py +++ b/docling/datamodel/document.py @@ -20,8 +20,30 @@ import filetype # DO NOT REMOVE; explicitly exposed from this location from docling_core.types.doc import ( + DocItem, DocItemLabel, DoclingDocument, + PictureItem, + SectionHeaderItem, + TableItem, + TextItem, +) +from docling_core.types.doc.document import ListItem +from docling_core.types.legacy_doc.base import ( + BaseText, + Figure, + GlmTableCell, + PageDimensions, + PageReference, + Prov, + Ref, + Table as DsSchemaTable, + TableCell, +) +from docling_core.types.legacy_doc.document import ( + CCSDocumentDescription as DsDocumentDescription, + CCSFileInfoObject as DsFileInfoObject, + ExportedCCSDocument as DsDocument, ) from docling_core.utils.file import resolve_source_to_stream from docling_core.utils.legacy import docling_document_to_legacy diff --git a/tests/test_backend_html.py b/tests/test_backend_html.py index 783137cc..18254a78 100644 --- a/tests/test_backend_html.py +++ b/tests/test_backend_html.py @@ -2,7 +2,6 @@ from io import BytesIO from pathlib import Path import pytest -from docling_core.types.doc.document import SectionHeaderItem from docling.backend.html_backend import HTMLDocumentBackend from docling.datamodel.base_models import InputFormat @@ -10,7 +9,7 @@ from docling.datamodel.document import ( ConversionResult, DoclingDocument, InputDocument, - # SectionHeaderItem, + SectionHeaderItem, ) from docling.document_converter import DocumentConverter diff --git a/tests/test_backend_msword.py b/tests/test_backend_msword.py index 2edf1734..3c1500ef 100644 --- a/tests/test_backend_msword.py +++ b/tests/test_backend_msword.py @@ -1,14 +1,12 @@ from pathlib import Path -from docling_core.types.doc.document import SectionHeaderItem - from docling.backend.msword_backend import MsWordDocumentBackend from docling.datamodel.base_models import InputFormat from docling.datamodel.document import ( ConversionResult, DoclingDocument, InputDocument, - # SectionHeaderItem, + SectionHeaderItem, ) from docling.document_converter import DocumentConverter