mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-09 13:18:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: align output formats (#49)

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi
2024-08-26 13:30:26 +02:00
committed by GitHub
parent 053eae4bdf
commit 8cc147bc56
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docling/document_converter.py
View File

@@ -88,7 +88,7 @@ class DocumentConverter:
# Note: Pdfium backend is not thread-safe, thread pool usage was disabled.
yield from map(self.process_document, input_batch)
def convert_single(self, source: Path | AnyHttpUrl | str) -> Document:
def convert_single(self, source: Path | AnyHttpUrl | str) -> ConvertedDocument:
"""Convert a single document.
Args:
@@ -133,11 +133,10 @@ class DocumentConverter:
converted_doc: ConvertedDocument = next(converted_docs_iter)
if converted_doc.status not in {
ConversionStatus.SUCCESS,
ConversionStatus.SUCCESS_WITH_ERRORS,
ConversionStatus.PARTIAL_SUCCESS,
}:
raise RuntimeError(f"Conversion failed with status: {converted_doc.status}")
doc = converted_doc.to_ds_document()
return doc
return converted_doc
def process_document(self, in_doc: InputDocument) -> ConvertedDocument:
start_doc_time = time.time()