diff --git a/spacy/pipeline/coref.py b/spacy/pipeline/coref.py index 489f8875b..ab315720f 100644 --- a/spacy/pipeline/coref.py +++ b/spacy/pipeline/coref.py @@ -561,9 +561,7 @@ class SpanPredictor(TrainablePipe): if key.startswith(self.output_prefix): for mention in sg: starts.append(mention.start) - # TODO check: Is the -1 here correct? - # In Akos's env it works without, but in Paul's it doesn't. - ends.append(mention.end - 1) + ends.append(mention.end) starts = self.model.ops.xp.asarray(starts) ends = self.model.ops.xp.asarray(ends)