From fce804a79f0b768516c054b703210f7b71cb9bbd Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Thu, 17 Jun 2021 21:10:46 +0900
Subject: [PATCH] Minor optimization

---
 spacy/ml/models/coref_util.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/spacy/ml/models/coref_util.py b/spacy/ml/models/coref_util.py
index cc79282d0..7ace3afcf 100644
--- a/spacy/ml/models/coref_util.py
+++ b/spacy/ml/models/coref_util.py
@@ -143,9 +143,12 @@ def get_candidate_mentions(
         si = sentence_map[tok.i]  # sentence index
         for ii in range(1, max_span_width):
             ei = tok.i + ii  # end index
-            if ei < len(doc) and sentence_map[ei] == si:
-                begins.append(tok.i)
-                ends.append(ei)
+
+            if ei > len(doc) or sentence_map[ei] != si:
+                continue
+
+            begins.append(tok.i)
+            ends.append(ei)
 
     return (begins, ends)