fix: fix metrics

Unstructured-IO · Dec 17, 2024 · 915bceb · 915bceb
1 parent 341073f
commit 915bceb
Showing 1 changed file with 4 additions and 6 deletions.
diff --git 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,