mirror of https://github.com/explosion/spaCy.git
return fix
This commit is contained in:
parent
a037206f0a
commit
4e7ec1ed31
|
@ -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]
|
||||||
|
|
Loading…
Reference in New Issue