explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-10 16:22:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in check for doc NEL viability.

Browse Source
This commit is contained in:
Raphael Mitsch 2022-10-23 18:32:55 +02:00
parent 9c69c3c313
commit f5f20b0b92
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/entity_linker.py
View File

@ -467,7 +467,7 @@ class EntityLinker(TrainablePipe):
if doc.ents[idx].label_ not in self.labels_discard
]
for doc in docs
if len(doc.ents)
if len(doc) and len(doc.ents)
)
# Call candidate generator.
if self.candidates_doc_mode:
@ -476,7 +476,7 @@ class EntityLinker(TrainablePipe):
(
[doc.ents[idx] for idx in next(valid_ent_idx_per_doc)]
for doc in docs
if len(docs) and len(doc.ents)
if len(doc) and len(doc.ents)
),
)
else:
@ -487,7 +487,7 @@ class EntityLinker(TrainablePipe):
for idx in next(valid_ent_idx_per_doc)
]
for doc in docs
if len(docs) and len(doc.ents)
if len(doc) and len(doc.ents)
)
for doc_idx, doc in enumerate(docs):