bugfix in qa span pred

allenai · Aug 5, 2023 · 5b4a5eb · 5b4a5eb
1 parent b23f2e4
commit 5b4a5eb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = 'papermage'
-version = '0.5.0'
+version = '0.5.1'
 description = 'Papermage. Casting magic over scientific PDFs.'
 license = {text = 'Apache-2.0'}
 readme = 'README.md'

diff --git a/tests/test_predictors/test_span_qa_predictor.py b/tests/test_predictors/test_span_qa_predictor.py
@@ -3,7 +3,7 @@
 import pathlib
 import unittest
 
-from papermage.magelib import Document, Entity, Span
+from papermage.magelib import Document, Entity, Metadata, Span
 from papermage.predictors.api_predictors.span_qa_predictor import APISpanQAPredictor
 
 
@@ -15,7 +15,7 @@ def setUp(self):
         self.doc = Document.from_json(doc_json=test_doc_json)
 
         user_selected_span = Entity(
-            spans=[Span(start=2784, end=2803)], metadata={"question": "What does this mean?"}
+            spans=[Span(start=2784, end=2803)], metadata=Metadata(question="What does this mean?")
         )
         self.doc.annotate_entity(field_name="user_selected_span", entities=[user_selected_span])