From 2b1e72d327311b5c361cc923df304aa9f42d46be Mon Sep 17 00:00:00 2001 From: Michele Dolfi <97102151+dolfim-ibm@users.noreply.github.com> Date: Mon, 14 Oct 2024 08:40:22 +0200 Subject: [PATCH] refactor: fix type of tesseractocr options (#140) Signed-off-by: Michele Dolfi <97102151+dolfim-ibm@users.noreply.github.com> --- docling/models/tesseract_ocr_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docling/models/tesseract_ocr_model.py b/docling/models/tesseract_ocr_model.py index 1b4f6f7..2b616ca 100644 --- a/docling/models/tesseract_ocr_model.py +++ b/docling/models/tesseract_ocr_model.py @@ -4,16 +4,16 @@ from typing import Iterable import numpy from docling.datamodel.base_models import BoundingBox, CoordOrigin, OcrCell, Page -from docling.datamodel.pipeline_options import TesseractCliOcrOptions +from docling.datamodel.pipeline_options import TesseractOcrOptions from docling.models.base_ocr_model import BaseOcrModel _log = logging.getLogger(__name__) class TesseractOcrModel(BaseOcrModel): - def __init__(self, enabled: bool, options: TesseractCliOcrOptions): + def __init__(self, enabled: bool, options: TesseractOcrOptions): super().__init__(enabled=enabled, options=options) - self.options: TesseractCliOcrOptions + self.options: TesseractOcrOptions self.scale = 3 # multiplier for 72 dpi == 216 dpi. self.reader = None