diff --git a/docling/cli/main.py b/docling/cli/main.py index ac7f289d..7f0f20bf 100644 --- a/docling/cli/main.py +++ b/docling/cli/main.py @@ -410,7 +410,7 @@ def convert( export_doctags = OutputFormat.DOCTAGS in to_formats ocr_factory = get_ocr_factory(allow_external_plugins=allow_external_plugins) - ocr_options: OcrOptions = ocr_factory.create_options( + ocr_options: OcrOptions = ocr_factory.create_options( # type: ignore kind=ocr_engine, force_full_page_ocr=force_ocr, ) diff --git a/docling/models/factories/__init__.py b/docling/models/factories/__init__.py index 6c7bc1bc..9a3308e1 100644 --- a/docling/models/factories/__init__.py +++ b/docling/models/factories/__init__.py @@ -10,7 +10,7 @@ logger = logging.getLogger(__name__) @lru_cache() -def get_ocr_factory(allow_external_plugins: bool = False): +def get_ocr_factory(allow_external_plugins: bool = False) -> OcrFactory: factory = OcrFactory() factory.load_from_plugins(allow_external_plugins=allow_external_plugins) logger.info("Registered ocr engines: %r", factory.registered_kind) @@ -18,7 +18,9 @@ def get_ocr_factory(allow_external_plugins: bool = False): @lru_cache() -def get_picture_description_factory(allow_external_plugins: bool = False): +def get_picture_description_factory( + allow_external_plugins: bool = False, +) -> PictureDescriptionFactory: factory = PictureDescriptionFactory() factory.load_from_plugins(allow_external_plugins=allow_external_plugins) logger.info("Registered picture descriptions: %r", factory.registered_kind)