return correct variable

This commit is contained in:
kadarakos 2023-05-31 17:30:17 +00:00
parent 6e46ecfa2c
commit f599bd5a4d

View File

@ -333,7 +333,7 @@ class SpanFinder(TrainablePipe):
masks.append(mask)
truths = ops.xp.concatenate(truths, axis=0)
masks = ops.xp.concatenate(masks, axis=0)
return truths, mask
return truths, masks
def _get_reference(self, docs) -> List[Tuple[int, int]]:
"""Create a reference list of token probabilities"""