This commit is contained in:
Raphael Mitsch 2023-03-10 14:21:23 +01:00
parent db73aedf51
commit 8c88543c0c

View File

@ -484,16 +484,13 @@ class EntityLinker(TrainablePipe):
# 2. "8 was beautiful" # 2. "8 was beautiful"
# 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].
if len(sents) > 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 (
) sentences.index(ent.sent),
else: sentences.index(ent.sent),
sent_indices = ( )
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: