From 915bceb2d6e0e444168b13e720cc8840962e21a4 Mon Sep 17 00:00:00 2001 From: Steve Canny Date: Tue, 17 Dec 2024 12:57:42 -0800 Subject: [PATCH] fix: fix metrics --- test_unstructured/metrics/test_element_type.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/test_unstructured/metrics/test_element_type.py b/test_unstructured/metrics/test_element_type.py index d1faba6a20..9a44a08f05 100644 --- a/test_unstructured/metrics/test_element_type.py +++ b/test_unstructured/metrics/test_element_type.py @@ -74,20 +74,18 @@ def test_get_element_type_frequency(filename: str, frequency: dict[tuple[str, in "handbook-1p.docx", { ("Header", None): 1, - ("Title", 0): 1, - ("Title", 1): 1, - ("Title", 2): 1, + ("UncategorizedText", 0): 6, ("ListItem", 3): 3, - ("NarrativeText", 4): 7, + ("NarrativeText", 0): 7, ("Footer", None): 1, }, - (0.43, 0.07, 0.65), + (0.78, 0.72, 0.81), ), ( "handbook-1p.docx", { ("Header", None): 1, - ("Title", 0): 6, + ("UncategorizedText", 0): 6, ("NarrativeText", 0): 7, ("PageBreak", None): 1, ("Footer", None): 1,