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

Format. Update assert in ent.sents handling.

Browse Source
This commit is contained in:
Raphael Mitsch 2023-03-14 13:18:11 +01:00
parent 9b6f47f1ac
commit fc0f954d9b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/pipeline/entity_linker.py
View File

@ -476,8 +476,11 @@ class EntityLinker(TrainablePipe):
for j, ent in enumerate(ent_batch): for j, ent in enumerate(ent_batch):
assert hasattr(ent, "sents") assert hasattr(ent, "sents")
sents = list(ent.sents) sents = list(ent.sents)
sent_indices = (sentences.index(sents[0]), sentences.index(sents[-1])) sent_indices = (
assert all([si >= 0 for si in sent_indices]) sentences.index(sents[0]),
sentences.index(sents[-1]),
)
assert sent_indices[-1] >= sent_indices[0] >= 0
if self.incl_context: if self.incl_context:
# get n_neighbour sentences, clipped to the length of the document # get n_neighbour sentences, clipped to the length of the document