deepdoctection 0.39.3__py3-none-any.whl → 0.39.4__py3-none-any.whl

deepdoctection/__init__.py

@@ -25,7 +25,7 @@ from .utils.logger import LoggingRecord, logger
 
 # pylint: enable=wrong-import-position
 
-__version__ = "0.39.3"
+__version__ = "0.39.4"
 
 _IMPORT_STRUCTURE = {
     "analyzer": ["config_sanity_checks", "get_dd_analyzer", "ServiceFactory"],

deepdoctection/extern/doctrocr.py

@@ -79,8 +79,8 @@ def _load_model(
         state_dict = torch.load(os.fspath(path_weights), map_location=device)
         for key in list(state_dict.keys()):
             state_dict["model." + key] = state_dict.pop(key)
-        doctr_predictor.load_state_dict(state_dict)
-        doctr_predictor.to(device)
+        doctr_predictor.load_state_dict(state_dict)  # type: ignore
+        doctr_predictor.to(device)  # type: ignore
     elif lib == "TF":
         # Unzip the archive
         params_path = Path(path_weights).parent
@@ -88,9 +88,9 @@ def _load_model(
         if is_zip_path:
             with ZipFile(path_weights, "r") as file:
                 file.extractall(path=params_path)
-                doctr_predictor.model.load_weights(params_path / "weights")
+                doctr_predictor.model.load_weights(params_path / "weights")  # type: ignore
         else:
-            doctr_predictor.model.load_weights(os.fspath(path_weights))
+            doctr_predictor.model.load_weights(os.fspath(path_weights))  # type: ignore
 
 
 def auto_select_lib_for_doctr() -> Literal["PT", "TF"]:
@@ -125,7 +125,7 @@ def doctr_predict_text_lines(
         DetectionResult(
             box=box[:4].tolist(), class_id=1, score=box[4], absolute_coords=False, class_name=LayoutType.WORD
         )
-        for box in raw_output[0]["words"]
+        for box in raw_output[0]["words"]  # type: ignore
     ]
     return detection_results
 
@@ -480,7 +480,7 @@ class DoctrTextRecognizer(TextRecognizer):
         return auto_select_lib_for_doctr()
 
     def clear_model(self) -> None:
-        self.doctr_predictor = None
+        self.doctr_predictor = None  # type: ignore
 
 
 class DocTrRotationTransformer(ImageTransformer):
@@ -527,7 +527,8 @@ class DocTrRotationTransformer(ImageTransformer):
         return viz_handler.rotate_image(np_img, specification.angle)  # type: ignore
 
     def predict(self, np_img: PixelValues) -> DetectionResult:
-        angle = estimate_orientation(np_img, self.number_contours, self.ratio_threshold_for_lines)
+        angle = estimate_orientation(np_img, n_ct=self.number_contours,
+                                     ratio_threshold_for_lines=self.ratio_threshold_for_lines)
         if angle < 0:
             angle += 360
         return DetectionResult(angle=round(angle, 2))

{deepdoctection-0.39.3.dist-info → deepdoctection-0.39.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: deepdoctection
-Version: 0.39.3
+Version: 0.39.4
 Summary: Repository for Document AI
 Home-page: https://github.com/deepdoctection/deepdoctection
 Author: Dr. Janis Meyer
@@ -57,7 +57,7 @@ Requires-Dist: tensorpack==0.11; extra == "tf"
 Requires-Dist: protobuf==3.20.1; extra == "tf"
 Requires-Dist: tensorflow-addons>=0.17.1; extra == "tf"
 Requires-Dist: tf2onnx>=1.9.2; extra == "tf"
-Requires-Dist: python-doctr==0.8.1; extra == "tf"
+Requires-Dist: python-doctr==0.9.0; extra == "tf"
 Requires-Dist: pycocotools>=2.0.2; extra == "tf"
 Requires-Dist: boto3==1.34.102; extra == "tf"
 Requires-Dist: pdfplumber>=0.11.0; extra == "tf"
@@ -88,7 +88,7 @@ Requires-Dist: tqdm==4.64.0; extra == "pt"
 Requires-Dist: timm>=0.9.16; extra == "pt"
 Requires-Dist: transformers>=4.48.0; extra == "pt"
 Requires-Dist: accelerate>=0.29.1; extra == "pt"
-Requires-Dist: python-doctr==0.8.1; extra == "pt"
+Requires-Dist: python-doctr==0.9.0; extra == "pt"
 Requires-Dist: boto3==1.34.102; extra == "pt"
 Requires-Dist: pdfplumber>=0.11.0; extra == "pt"
 Requires-Dist: fasttext-wheel; extra == "pt"

{deepdoctection-0.39.3.dist-info → deepdoctection-0.39.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-deepdoctection/__init__.py,sha256=ESGz9jn4mQi0T_Il2gIS6CArPLUrfANzgIQLHwyw8eE,12754
+deepdoctection/__init__.py,sha256=4u3rGxdbvfFr1l6xzH6qO5MapLGspBv5XPhBGMeji_4,12754
 deepdoctection/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 deepdoctection/analyzer/__init__.py,sha256=icClxrd20XutD6LxLgEPIWceSs4j_QfI3szCE-9BL2w,729
 deepdoctection/analyzer/_config.py,sha256=1rfvVrp7cI2YLzpahD77aa1tZ_KFAIQ21DM1NWhxYiI,5058
@@ -53,7 +53,7 @@ deepdoctection/extern/__init__.py,sha256=9Iks9b4Q_LynjcV167TVCoK8YsQRUcA2jjmAmDN
 deepdoctection/extern/base.py,sha256=ONPgappl_P5HSwQr42FatuRnwMTvUPecPsCztDTN0Hw,24108
 deepdoctection/extern/d2detect.py,sha256=zrKv1yurApnjD7QZIZk_8LYCahjmN82MQUjHjv8zvkQ,22127
 deepdoctection/extern/deskew.py,sha256=sPoixu8S9he-0wbs-jgxtPE2V9BiP4-3uZlb6F5Y1SA,3077
-deepdoctection/extern/doctrocr.py,sha256=T3_tvlih22_dVCBZypS1Y8tjQQB1fkAxIbGdUGHIapQ,24473
+deepdoctection/extern/doctrocr.py,sha256=WrFA0N54fr9C_ahGzZJb4H-fNzz5wXQFveFiERYAm74,24637
 deepdoctection/extern/fastlang.py,sha256=F4gK-SEwcCujjxH327ZDzMGWToJ49xS_dCKcePQ9IlY,4780
 deepdoctection/extern/hfdetr.py,sha256=JzHrrTyzS9qh6T2TsvKboAGZkIhno2txmSoLQ5Vd-lo,12077
 deepdoctection/extern/hflayoutlm.py,sha256=tFaf90FRbZzhSycdp8rGkeiPywQa6UcTEEwbayIXkr0,57023
@@ -141,8 +141,8 @@ deepdoctection/utils/transform.py,sha256=3kCgsEeRkG1efCdkfvj7tUFMs-e2jbjbflq826F
 deepdoctection/utils/types.py,sha256=_3dmPdCIZNLbgU5QP5k_c5phDf18xLe1kYL6t2nM45s,2953
 deepdoctection/utils/utils.py,sha256=csVs_VvCq4QBETPoE2JdTTL4MFYnD4xh-Js5vRb612g,6492
 deepdoctection/utils/viz.py,sha256=Jf8ePNYWlpuyaS6SeTYQ4OyA3eNhtgjvAQZnGNdgHC0,27051
-deepdoctection-0.39.3.dist-info/LICENSE,sha256=GQ0rUvuGdrMNEI3iHK5UQx6dIMU1QwAuyXsxUHn5MEQ,11351
-deepdoctection-0.39.3.dist-info/METADATA,sha256=CCHsl3E3U1slBqAfbxiGChEDfDH8rJRRGmLhz56LuLc,19741
-deepdoctection-0.39.3.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-deepdoctection-0.39.3.dist-info/top_level.txt,sha256=hs2DdoOL9h4mnHhmO82BT4pz4QATIoOZ20PZmlnxFI8,15
-deepdoctection-0.39.3.dist-info/RECORD,,
+deepdoctection-0.39.4.dist-info/LICENSE,sha256=GQ0rUvuGdrMNEI3iHK5UQx6dIMU1QwAuyXsxUHn5MEQ,11351
+deepdoctection-0.39.4.dist-info/METADATA,sha256=9DvyyvO9Ofd9YFuqQ2SVO2Bi69WayBC2VZyrpMCimZQ,19741
+deepdoctection-0.39.4.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+deepdoctection-0.39.4.dist-info/top_level.txt,sha256=hs2DdoOL9h4mnHhmO82BT4pz4QATIoOZ20PZmlnxFI8,15
+deepdoctection-0.39.4.dist-info/RECORD,,