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

Harmonize checks for entity candidate generation and document skipping.

Browse Source
This commit is contained in:
Raphael Mitsch 2022-10-21 13:14:03 +02:00
parent b32f48c878
commit 9c69c3c313
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

@ -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(doc.ents)
if len(docs) and len(doc.ents)
),
)
else:
@ -487,11 +487,11 @@ class EntityLinker(TrainablePipe):
for idx in next(valid_ent_idx_per_doc)
]
for doc in docs
if len(doc.ents)
if len(docs) and len(doc.ents)
)
for doc_idx, doc in enumerate(docs):
if len(doc) == 0:
if len(doc) == 0 or len(doc.ents) == 0:
continue
sentences = [s for s in doc.sents]
doc_ent_cands = list(next(all_ent_cands)) if len(doc.ents) else []