return fix

This commit is contained in:
svlandeg 2019-07-23 14:23:58 +02:00
parent a037206f0a
commit 4e7ec1ed31
1 changed files with 1 additions and 1 deletions

View File

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