diff --git a/CHANGELOG.md b/CHANGELOG.md index a3fb7ab6..e9b58ffd 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,7 @@ +## 1.5.3 + +- Store routing in LayoutElement + ## 1.5.2 ### Fix diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py index 928be505..e375e78d 100644 --- a/unstructured_inference/__version__.py +++ b/unstructured_inference/__version__.py @@ -1 +1 @@ -__version__ = "1.5.2" # pragma: no cover +__version__ = "1.5.3" # pragma: no cover diff --git a/unstructured_inference/inference/layout.py b/unstructured_inference/inference/layout.py index 5b956d2a..25c510ad 100644 --- a/unstructured_inference/inference/layout.py +++ b/unstructured_inference/inference/layout.py @@ -203,9 +203,13 @@ def get_elements_with_detection_model( # remote call in the future. assert self.image is not None inferred_layout: LayoutElements = self.detection_model(self.image) + routing = inferred_layout.routing + routing_score = inferred_layout.routing_score inferred_layout = self.detection_model.deduplicate_detected_elements( inferred_layout, ) + inferred_layout.routing = routing + inferred_layout.routing_score = routing_score if inplace: self.elements_array = inferred_layout diff --git a/unstructured_inference/inference/layoutelement.py b/unstructured_inference/inference/layoutelement.py index fb3e073e..23f87094 100644 --- a/unstructured_inference/inference/layoutelement.py +++ b/unstructured_inference/inference/layoutelement.py @@ -26,6 +26,8 @@ class LayoutElements(TextRegions): element_class_id_map: dict[int, str] = field(default_factory=dict) text_as_html: np.ndarray = field(default_factory=lambda: np.array([])) table_as_cells: np.ndarray = field(default_factory=lambda: np.array([])) + routing: str | None = None + routing_score: float | None = None _optional_array_attributes: list[str] = field( init=False, default_factory=lambda: [