From ff3fed06cf5733647122718a0bf7d842a1d497a0 Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Thu, 20 May 2021 21:30:46 +0900
Subject: [PATCH] Catch a stray reference

---
 spacy/ml/models/coref.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/spacy/ml/models/coref.py b/spacy/ml/models/coref.py
index 0f544ed2b..d4a28c0e4 100644
--- a/spacy/ml/models/coref.py
+++ b/spacy/ml/models/coref.py
@@ -391,6 +391,7 @@ def ant_scorer_forward(
     vecshape = vecs.data.shape
     veclens = vecs.lengths
     scoreshape = mscores.shape
+    idxes = sembeds.indices
 
     def backprop(
         dYs: Tuple[List[Tuple[Floats2d, Ints2d]], Ints2d]
@@ -417,7 +418,7 @@ def ant_scorer_forward(
             offset += ll
         # make it fit back into the linear
         dXscores = xp.expand_dims(dXscores, 1)
-        return (dXscores, SpanEmbeddings(sembeds.indices, dXembeds))
+        return (dXscores, SpanEmbeddings(idxes, dXembeds))
 
     return (out, sembeds.indices), backprop