From ba521dd88f52fa3297e753dbc4fe1295df8ea879 Mon Sep 17 00:00:00 2001 From: Panos Vagenas <35837085+vagenas@users.noreply.github.com> Date: Tue, 28 Jan 2025 16:17:44 +0100 Subject: [PATCH] chore: add missing imports to Office type tests (#826) * chore: add missing import to XLSX test Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com> * Update test_backend_msword.py [skip ci] Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com> * Update test_backend_pptx.py Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com> --------- Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com> --- tests/test_backend_msexcel.py | 2 +- tests/test_backend_msword.py | 1 + tests/test_backend_pptx.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_backend_msexcel.py b/tests/test_backend_msexcel.py index e664ed3..f33dffa 100644 --- a/tests/test_backend_msexcel.py +++ b/tests/test_backend_msexcel.py @@ -3,7 +3,7 @@ import os from pathlib import Path from docling.datamodel.base_models import InputFormat -from docling.datamodel.document import ConversionResult +from docling.datamodel.document import ConversionResult, DoclingDocument from docling.document_converter import DocumentConverter GENERATE = False diff --git a/tests/test_backend_msword.py b/tests/test_backend_msword.py index 24db677..9edcb3e 100644 --- a/tests/test_backend_msword.py +++ b/tests/test_backend_msword.py @@ -6,6 +6,7 @@ from docling.backend.msword_backend import MsWordDocumentBackend from docling.datamodel.base_models import InputFormat from docling.datamodel.document import ( ConversionResult, + DoclingDocument, InputDocument, SectionHeaderItem, ) diff --git a/tests/test_backend_pptx.py b/tests/test_backend_pptx.py index 4c3872b..f4799a8 100644 --- a/tests/test_backend_pptx.py +++ b/tests/test_backend_pptx.py @@ -3,7 +3,7 @@ import os from pathlib import Path from docling.datamodel.base_models import InputFormat -from docling.datamodel.document import ConversionResult +from docling.datamodel.document import ConversionResult, DoclingDocument from docling.document_converter import DocumentConverter GENERATE = False