Fix naming of offset in answers of TransformersReader (for consistenc…

…y with FARMReader) (#204)
deepset-ai · Jul 7, 2020 · 9733e0b · 9733e0b
1 parent b4f4388
commit 9733e0b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/haystack/reader/transformers.py b/haystack/reader/transformers.py
@@ -84,8 +84,8 @@ def predict(self, question: str, documents: List[Document], top_k: Optional[int]
                     answers.append({
                         "answer": pred["answer"],
                         "context": doc.text[context_start:context_end],
-                        "offset_answer_start": pred["start"],
-                        "offset_answer_end": pred["end"],
+                        "offset_start": pred["start"],
+                        "offset_end": pred["end"],
                         "probability": pred["score"],
                         "score": None,
                         "document_id": doc.id,