deepdoctection 0.39.2__tar.gz → 0.39.3__tar.gz

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: deepdoctection
-Version: 0.39.2
+Version: 0.39.3
 Summary: Repository for Document AI
 Home-page: https://github.com/deepdoctection/deepdoctection
 Author: Dr. Janis Meyer

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection/__init__.py

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

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection/datapoint/image.py

@@ -702,11 +702,11 @@ class Image:
         return get_uuid(self.image_id, *container_ids)
 
     def save(
-        self,
-        image_to_json: bool = True,
-        highest_hierarchy_only: bool = False,
-        path: Optional[PathLikeOrStr] = None,
-        dry: bool = False,
+            self,
+            image_to_json: bool = True,
+            highest_hierarchy_only: bool = False,
+            path: Optional[PathLikeOrStr] = None,
+            dry: bool = False,
     ) -> Optional[Union[ImageDict, str]]:
         """
         Export image as dictionary. As numpy array cannot be serialized `image` values will be converted into
@@ -719,6 +719,18 @@ class Image:
 
         :return: optional dict
         """
+
+        def set_image_keys_to_none(d):  # type: ignore
+            if isinstance(d, dict):
+                for key, value in d.items():
+                    if key == '_image':
+                        d[key] = None
+                    else:
+                        set_image_keys_to_none(value)
+            elif isinstance(d, list):
+                for item in d:
+                    set_image_keys_to_none(item)
+
         if path is None:
             path = Path(self.location)
         path = Path(path)
@@ -734,7 +746,7 @@ class Image:
         export_dict = self.as_dict()
         export_dict["location"] = fspath(export_dict["location"])
         if not image_to_json:
-            export_dict["_image"] = None
+            set_image_keys_to_none(export_dict)
         if dry:
             return export_dict
         with open(path_json, "w", encoding="UTF-8") as file:

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection/datasets/registry.py

@@ -18,6 +18,7 @@
 """
 Module for DatasetRegistry
 """
+import inspect
 
 import catalogue  # type: ignore
 from tabulate import tabulate
@@ -47,7 +48,10 @@ def get_dataset(name: str) -> DatasetBase:
     :param name: A dataset name
     :return: An instance of a dataset
     """
-    return dataset_registry.get(name)()
+    ds = dataset_registry.get(name)
+    if inspect.isclass(ds):
+        return ds()
+    return ds
 
 
 def print_dataset_infos(add_license: bool = True, add_info: bool = True) -> None:

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection/mapper/cats.py

@@ -32,7 +32,7 @@ from .maputils import LabelSummarizer, curry
 @curry
 def cat_to_sub_cat(
     dp: Image,
-    categories_dict_names_as_key: dict[TypeOrStr, int],
+    categories_dict_names_as_key: Optional[dict[TypeOrStr, int]] = None,
     cat_to_sub_cat_dict: Optional[dict[TypeOrStr, TypeOrStr]] = None,
 ) -> Image:
     """
@@ -45,7 +45,8 @@ def cat_to_sub_cat(
     :param cat_to_sub_cat_dict: e.g. {'foo': 'sub_cat_1', 'bak': 'sub_cat_2'}
     :return: Image with updated Annotations
     """
-
+    if categories_dict_names_as_key is None:
+        categories_dict_names_as_key = {}
     if cat_to_sub_cat_dict is None:
         return dp
     cat_to_sub_cat_dict_obj_type = {get_type(key): get_type(value) for key, value in cat_to_sub_cat_dict.items()}
@@ -54,7 +55,7 @@ def cat_to_sub_cat(
         sub_cat = ann.get_sub_category(sub_cat_type)
         if sub_cat:
             ann.category_name = sub_cat.category_name
-            ann.category_id = categories_dict_names_as_key[ann.category_name]
+            ann.category_id = categories_dict_names_as_key.get(ann.category_name,DEFAULT_CATEGORY_ID)
 
     return dp
 

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection/pipe/segment.py

@@ -1190,14 +1190,16 @@ class PubtablesSegmentationService(PipelineComponent):
                             for key, value in cell_rn_cn_to_ann_id.items():
                                 if key[idx] == item_number:
                                     cell_ann = dp.get_annotation(annotation_ids=value)[0]
-                                    self.dp_manager.set_category_annotation(
-                                        item_header_cell_name, None, item_header_cell_name, cell_ann.annotation_id
-                                    )
+                                    if item_header_cell_name not in cell_ann.sub_categories:
+                                        self.dp_manager.set_category_annotation(
+                                            item_header_cell_name, None, item_header_cell_name, cell_ann.annotation_id
+                                        )
                                 else:
                                     cell_ann = dp.get_annotation(annotation_ids=value)[0]
-                                    self.dp_manager.set_category_annotation(
-                                        item_header_cell_name, None, CellType.BODY, cell_ann.annotation_id
-                                    )
+                                    if CellType.BODY not in cell_ann.sub_categories:
+                                        self.dp_manager.set_category_annotation(
+                                            item_header_cell_name, None, CellType.BODY, cell_ann.annotation_id
+                                        )
 
             # TODO: the summaries should be sub categories of the underlying ann
             self.dp_manager.set_summary_annotation(

{deepdoctection-0.39.2 → deepdoctection-0.39.3}/deepdoctection.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: deepdoctection
-Version: 0.39.2
+Version: 0.39.3
 Summary: Repository for Document AI
 Home-page: https://github.com/deepdoctection/deepdoctection
 Author: Dr. Janis Meyer