From 4fe98f1a0a7da882edbe9c6be1fce0f32e918743 Mon Sep 17 00:00:00 2001 From: Panos Vagenas <35837085+vagenas@users.noreply.github.com> Date: Tue, 15 Oct 2024 10:11:29 +0200 Subject: [PATCH] switch convert_all output type from Iterable to Iterator Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com> --- docling/document_converter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docling/document_converter.py b/docling/document_converter.py index d2324e40..c3bc2dac 100644 --- a/docling/document_converter.py +++ b/docling/document_converter.py @@ -3,7 +3,7 @@ import sys import time from functools import partial from pathlib import Path -from typing import Dict, Iterable, List, Optional, Type +from typing import Dict, Iterable, Iterator, List, Optional, Type from pydantic import BaseModel, ConfigDict, model_validator, validate_call @@ -137,7 +137,7 @@ class DocumentConverter: raises_on_error: bool = True, # True: raises on first conversion error; False: does not raise on conv error max_num_pages: int = sys.maxsize, max_file_size: int = sys.maxsize, - ) -> Iterable[ConversionResult]: + ) -> Iterator[ConversionResult]: limits = DocumentLimits( max_num_pages=max_num_pages, max_file_size=max_file_size, @@ -160,7 +160,7 @@ class DocumentConverter: def _convert( self, conv_input: _DocumentConversionInput, raises_on_error: bool - ) -> Iterable[ConversionResult]: + ) -> Iterator[ConversionResult]: for input_batch in chunkify( conv_input.docs(self.format_to_options), settings.perf.doc_batch_size, # pass format_options