From f599bd5a4dde8d4be2bccb2f4bb67f6e1789bb5d Mon Sep 17 00:00:00 2001 From: kadarakos Date: Wed, 31 May 2023 17:30:17 +0000 Subject: [PATCH] return correct variable --- spacy/pipeline/span_finder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/pipeline/span_finder.py b/spacy/pipeline/span_finder.py index 50112d692..5b6f25262 100644 --- a/spacy/pipeline/span_finder.py +++ b/spacy/pipeline/span_finder.py @@ -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"""