From 4772fd17e2fd55e5b8b83490a2210ac32f9a73ab Mon Sep 17 00:00:00 2001 From: kba Date: Wed, 29 Oct 2025 20:47:13 +0100 Subject: [PATCH] missed changing override mechanism in eynollah_ocr --- src/eynollah/eynollah_ocr.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/src/eynollah/eynollah_ocr.py b/src/eynollah/eynollah_ocr.py index d32777a..0f3eda6 100644 --- a/src/eynollah/eynollah_ocr.py +++ b/src/eynollah/eynollah_ocr.py @@ -1,4 +1,9 @@ +# FIXME: fix all of those... # pyright: reportPossiblyUnboundVariable=false +# pyright: reportOptionalMemberAccess=false +# pyright: reportArgumentType=false +# pyright: reportCallIssue=false +# pyright: reportOptionalSubscript=false from logging import Logger, getLogger from typing import Optional @@ -53,7 +58,6 @@ class Eynollah_ocr: self, *, model_zoo: EynollahModelZoo, - dir_xmls=None, tr_ocr=False, batch_size: Optional[int]=None, export_textline_images_and_text: bool=False, @@ -82,12 +86,13 @@ class Eynollah_ocr: if tr_ocr: self.model_zoo.load_model('trocr_processor') - self.model_zoo.load_model('ocr', 'tr', model_path_override=model_name) + self.model_zoo.load_model('ocr', 'tr') self.model_zoo.get('ocr').to(self.device) else: - self.model_zoo.load_model('ocr', '', model_path_override=model_name) + self.model_zoo.load_model('ocr', '') self.model_zoo.load_model('num_to_char') - self.end_character = len(self.model_zoo.load_model('characters')) + 2 + self.model_zoo.load_model('characters') + self.end_character = len(self.model_zoo.get('characters', list)) + 2 @property def device(self):