use the string representation of the abstract and section name annota…

…tions
ArneBinder · Dec 15, 2023 · 9e0fe9e · 9e0fe9e
1 parent 8f5f3fe
commit 9e0fe9e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/dataset_builders/pie/test_scientific_papers.py b/tests/dataset_builders/pie/test_scientific_papers.py
@@ -108,9 +108,10 @@ def test_hf_example(hf_example, expected_output):
 def test_pie_example(pie_example, expected_output):
     assert pie_example is not None
     assert pie_example.text.startswith(expected_output["article"])
-    assert pie_example.abstract[0].text.startswith(expected_output["abstract"])
+    # Note that we use the string representation of the abstract and section name annotations
+    assert str(pie_example.abstract[0]).startswith(expected_output["abstract"])
     str_section_names = "\n".join(
-        [section_name.text for section_name in pie_example.section_names]
+        [str(section_name) for section_name in pie_example.section_names]
     )
     assert str_section_names.startswith(expected_output["section_names"])