test_input_doc use default backend
Some checks failed
Run Docs CI / build-docs (push) Failing after 1m27s
Run CI / code-checks (push) Failing after 6m38s

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-03-17 11:13:42 +01:00
parent eff907811a
commit 50ac62b5fa

View File

@ -8,6 +8,7 @@ from docling.backend.pypdfium2_backend import PyPdfiumDocumentBackend
from docling.datamodel.base_models import DocumentStream, InputFormat from docling.datamodel.base_models import DocumentStream, InputFormat
from docling.datamodel.document import InputDocument, _DocumentConversionInput from docling.datamodel.document import InputDocument, _DocumentConversionInput
from docling.datamodel.settings import DocumentLimits from docling.datamodel.settings import DocumentLimits
from docling.document_converter import PdfFormatOption
def test_in_doc_from_valid_path(): def test_in_doc_from_valid_path():
@ -227,7 +228,7 @@ def _make_input_doc(path):
in_doc = InputDocument( in_doc = InputDocument(
path_or_stream=path, path_or_stream=path,
format=InputFormat.PDF, format=InputFormat.PDF,
backend=DoclingParseV4DocumentBackend, backend=PdfFormatOption().backend, # use default
) )
return in_doc return in_doc
@ -237,6 +238,6 @@ def _make_input_doc_from_stream(doc_stream):
path_or_stream=doc_stream.stream, path_or_stream=doc_stream.stream,
format=InputFormat.PDF, format=InputFormat.PDF,
filename=doc_stream.name, filename=doc_stream.name,
backend=DoclingParseV4DocumentBackend, backend=PdfFormatOption().backend, # use default
) )
return in_doc return in_doc