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

Format.

Browse Source
This commit is contained in:
Raphael Mitsch 2023-03-10 14:21:31 +01:00
parent 8c88543c0c
commit e64a9ea499
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/entity_linker.py
View File

@ -485,12 +485,16 @@ class EntityLinker(TrainablePipe):
# This makes it tricky to receive the last sentence by indexing doc.sents - hence we use an offset # This makes it tricky to receive the last sentence by indexing doc.sents - hence we use an offset
# to determine sent_indices[1]. # to determine sent_indices[1].
sent_indices = ( sent_indices = (
(
sentences.index(sents[0]), sentences.index(sents[0]),
sentences.index(sents[0]) + len(sents) - 1, sentences.index(sents[0]) + len(sents) - 1,
) if len(sents) > 1 else ( )
if len(sents) > 1
else (
sentences.index(ent.sent), sentences.index(ent.sent),
sentences.index(ent.sent), sentences.index(ent.sent),
) )
)
assert all([si >= 0 for si in sent_indices]) assert all([si >= 0 for si in sent_indices])
if self.incl_context: if self.incl_context: