From 8c08c204891508b2e6f768cf3f038ba31341c6fd Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 20:52:23 -0700 Subject: [PATCH 1/6] fix maybe --- CHANGELOG.md | 3 +++ unstructured_inference/__version__.py | 2 +- unstructured_inference/inference/layoutelement.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 48679596..d5c5f8b3 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,6 @@ +## 0.8.1 +* fix: fix a bug caused by making LayoutElements subject from empty page + ## 0.8.0 * fix: fix missing source after cleaning layout elements diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py index f7ca6430..6719f022 100644 --- a/unstructured_inference/__version__.py +++ b/unstructured_inference/__version__.py @@ -1 +1 @@ -__version__ = "0.8.0" # pragma: no cover +__version__ = "0.8.1" # pragma: no cover diff --git a/unstructured_inference/inference/layoutelement.py b/unstructured_inference/inference/layoutelement.py index 332df603..f1b40ab4 100644 --- a/unstructured_inference/inference/layoutelement.py +++ b/unstructured_inference/inference/layoutelement.py @@ -145,7 +145,7 @@ def from_list(cls, elements: list): element_probs=np.array(class_probs), element_class_ids=class_ids, element_class_id_map=dict(zip(range(len(unique_ids)), unique_ids)), - source=elements[0].source, + source=elements[0].source if len_ele else None, ) From b528d684de659a100de20c5b104bff1367376763 Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 21:07:13 -0700 Subject: [PATCH 2/6] tidy test --- test_unstructured_inference/test_elements.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/test_unstructured_inference/test_elements.py b/test_unstructured_inference/test_elements.py index 7b70d35d..f061623e 100644 --- a/test_unstructured_inference/test_elements.py +++ b/test_unstructured_inference/test_elements.py @@ -436,3 +436,8 @@ def test_layoutelements_to_list_and_back(test_layoutelements): test_layoutelements.element_class_id_map[idx] for idx in test_layoutelements.element_class_ids ] == [back.element_class_id_map[idx] for idx in back.element_class_ids] + +def test_layoutelements_from_list_no_elements(): + back = LayoutElements.from_list(elements=[]) + assert back.source is None + \ No newline at end of file From 52b53503f36b5227d05ae292ee6e5c52f43a6bba Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 21:07:34 -0700 Subject: [PATCH 3/6] lint --- test_unstructured_inference/test_elements.py | 1 - 1 file changed, 1 deletion(-) diff --git a/test_unstructured_inference/test_elements.py b/test_unstructured_inference/test_elements.py index f061623e..a26d7d5b 100644 --- a/test_unstructured_inference/test_elements.py +++ b/test_unstructured_inference/test_elements.py @@ -440,4 +440,3 @@ def test_layoutelements_to_list_and_back(test_layoutelements): def test_layoutelements_from_list_no_elements(): back = LayoutElements.from_list(elements=[]) assert back.source is None - \ No newline at end of file From 5a9a4080a3e76c47b5ca30c481cc699b39084dd0 Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 21:07:59 -0700 Subject: [PATCH 4/6] lint --- test_unstructured_inference/test_elements.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test_unstructured_inference/test_elements.py b/test_unstructured_inference/test_elements.py index a26d7d5b..cb730335 100644 --- a/test_unstructured_inference/test_elements.py +++ b/test_unstructured_inference/test_elements.py @@ -437,6 +437,7 @@ def test_layoutelements_to_list_and_back(test_layoutelements): for idx in test_layoutelements.element_class_ids ] == [back.element_class_id_map[idx] for idx in back.element_class_ids] + def test_layoutelements_from_list_no_elements(): back = LayoutElements.from_list(elements=[]) assert back.source is None From bbd9f949f85b0282598c8d62a9b4dcdeaa5e2a00 Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 21:09:07 -0700 Subject: [PATCH 5/6] changelog entry nit --- CHANGELOG.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index d5c5f8b3..384b552a 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,5 @@ ## 0.8.1 -* fix: fix a bug caused by making LayoutElements subject from empty page +* fix: fix a bug caused by calling LayoutElements.from_list() with empty list ## 0.8.0 From def27637f2246adeb5ad5367881348a9873928ea Mon Sep 17 00:00:00 2001 From: Yuming Long Date: Thu, 24 Oct 2024 21:09:44 -0700 Subject: [PATCH 6/6] more specific --- CHANGELOG.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 384b552a..75673742 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,5 @@ ## 0.8.1 -* fix: fix a bug caused by calling LayoutElements.from_list() with empty list +* fix: fix list index out of range error caused by calling LayoutElements.from_list() with empty list ## 0.8.0