Skip to content

Commit 82fd0d0

Browse files
committed
fix another possible crash in the model output parsing
1 parent 035e85a commit 82fd0d0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tibert/bertcoref.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -902,7 +902,10 @@ def coreference_documents(
902902
antecedent_span_i = int(
903903
self.top_antecedents_index[b_i][m_j][top_antecedent_idx].item()
904904
)
905-
antecedent_coords = spans_idx[antecedent_span_i]
905+
try:
906+
antecedent_coords = spans_idx[antecedent_span_i]
907+
except IndexError:
908+
continue
906909

907910
antecedent_mention_score = float(
908911
self.mentions_scores[b_i][antecedent_span_i].item()

0 commit comments

Comments
 (0)