explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-06 21:30:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use .sum() as in update

Browse Source
This commit is contained in:
Adriane Boyd 2022-11-29 17:18:55 +01:00
parent 4ecccf8460
commit 965bbb3e0d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/spancat.py
View File

@ -272,7 +272,7 @@ class SpanCategorizer(TrainablePipe):
DOCS: https://spacy.io/api/spancategorizer#predict DOCS: https://spacy.io/api/spancategorizer#predict
""" """
indices = self.suggester(docs, ops=self.model.ops) indices = self.suggester(docs, ops=self.model.ops)
if sum(indices.lengths) == 0: if indices.lengths.sum() == 0:
scores = self.model.ops.alloc2f(0, 0) scores = self.model.ops.alloc2f(0, 0)
else: else:
scores = self.model.predict((docs, indices)) # type: ignore scores = self.model.predict((docs, indices)) # type: ignore