Remove end adjustment

The difference in environments was due to a change in Thinc, the code
here is fine.
This commit is contained in:
Paul O'Leary McCann 2022-04-14 18:31:30 +09:00
parent 8181d4570c
commit 08729e0fbd

View File

@ -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)