refactor: fix type of tesseractocr options (#140)

Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Oct 14, 2024 · 2b1e72d · 2b1e72d
1 parent 4672b24
commit 2b1e72d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docling/models/tesseract_ocr_model.py b/docling/models/tesseract_ocr_model.py
@@ -4,16 +4,16 @@
 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