mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-10 16:22:29 +03:00
return correct variable
This commit is contained in:
parent
6e46ecfa2c
commit
f599bd5a4d
|
@ -333,7 +333,7 @@ class SpanFinder(TrainablePipe):
|
||||||
masks.append(mask)
|
masks.append(mask)
|
||||||
truths = ops.xp.concatenate(truths, axis=0)
|
truths = ops.xp.concatenate(truths, axis=0)
|
||||||
masks = ops.xp.concatenate(masks, axis=0)
|
masks = ops.xp.concatenate(masks, axis=0)
|
||||||
return truths, mask
|
return truths, masks
|
||||||
|
|
||||||
def _get_reference(self, docs) -> List[Tuple[int, int]]:
|
def _get_reference(self, docs) -> List[Tuple[int, int]]:
|
||||||
"""Create a reference list of token probabilities"""
|
"""Create a reference list of token probabilities"""
|
||||||
|
|
Loading…
Reference in New Issue
Block a user