From 753f67a434960cbc41cffa429e194e5903ef3b91 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Fri, 11 Oct 2024 13:06:32 +0200 Subject: [PATCH] fixes Signed-off-by: Michele Dolfi --- docling/models/page_preprocessing_model.py | 4 ++-- docling/pipeline/standard_pdf_model_pipeline.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docling/models/page_preprocessing_model.py b/docling/models/page_preprocessing_model.py index 438114d8..52f92129 100644 --- a/docling/models/page_preprocessing_model.py +++ b/docling/models/page_preprocessing_model.py @@ -1,4 +1,4 @@ -from typing import Iterable +from typing import Iterable, Optional from PIL import ImageDraw from pydantic import BaseModel @@ -8,7 +8,7 @@ from docling.models.abstract_model import AbstractPageModel class PagePreprocessingOptions(BaseModel): - images_scale: float + images_scale: Optional[float] class PagePreprocessingModel(AbstractPageModel): diff --git a/docling/pipeline/standard_pdf_model_pipeline.py b/docling/pipeline/standard_pdf_model_pipeline.py index cba8609b..53558579 100644 --- a/docling/pipeline/standard_pdf_model_pipeline.py +++ b/docling/pipeline/standard_pdf_model_pipeline.py @@ -47,7 +47,7 @@ class StandardPdfModelPipeline(PaginatedModelPipeline): ) ) - if ocr_model := self.get_ocr_model() is None: + if (ocr_model := self.get_ocr_model()) is None: raise RuntimeError( f"The specified OCR kind is not supported: {pipeline_options.ocr_options.kind}." )