return fix

This commit is contained in:
svlandeg 2019-07-23 14:23:58 +02:00
parent a037206f0a
commit 4e7ec1ed31

View File

@ -1237,7 +1237,7 @@ class EntityLinker(Pipe):
final_tensors = []
if not docs:
return final_kb_ids
return final_kb_ids, final_tensors
if isinstance(docs, Doc):
docs = [docs]