From e847487ebb2f663fbbe7442e077beaa617bb0774 Mon Sep 17 00:00:00 2001 From: kadarakos Date: Mon, 20 Feb 2023 10:36:54 +0000 Subject: [PATCH] remove duplicate declaration --- spacy/pipeline/spancat.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spacy/pipeline/spancat.py b/spacy/pipeline/spancat.py index 1de51bb09..2f10a2ff4 100644 --- a/spacy/pipeline/spancat.py +++ b/spacy/pipeline/spancat.py @@ -712,9 +712,8 @@ class SpanCategorizer(TrainablePipe): ) -> SpanGroup: """Find the argmax label for each span.""" # Handle cases when there are zero suggestions - spans = SpanGroup(doc, name=self.key) if scores.size == 0: - return spans + return SpanGroup(doc, name=self.key) scores = self.model.ops.to_numpy(scores) indices = self.model.ops.to_numpy(indices) predicted = scores.argmax(axis=1)