yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-07 20:46:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix document_name for TFIDF retriever (#33)

Browse Source
This commit is contained in:
Malte Pietsch 2020-03-01 16:08:50 +01:00 committed by GitHub
parent a0cfd16b0c
commit f1a9dad24f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
haystack/finder.py
View File

@ -42,9 +42,9 @@ class Finder:
# Add corresponding document_name if an answer contains the document_id (only supported in FARMReader)
for ans in results["answers"]:
document_name = next(
(meta["document_name"] for meta in meta_data if meta["document_id"] == ans["document_id"]), None
)
ans["document_name"] = document_name
ans["document_name"] = None
for meta in meta_data:
if meta["document_id"] == ans["document_id"]:
ans["document_name"] = meta.get("document_name", None)
return results