explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-24 19:11:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

Remove mypy ignore and typecast labels to list

Browse Source
This commit is contained in:
Lj Miranda 2022-11-29 11:14:43 +08:00
parent 14ae4a52c0
commit 0b32a949f1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/spancat_exclusive.py
View File

@ -191,7 +191,7 @@ class Exclusive_SpanCategorizer(SpanCategorizer):
DOCS: https://spacy.io/api/spancatcategorizer#set_annotations
"""
allow_overlap = cast(bool, self.cfg["allow_overlap"])
labels = self.labels
labels = list(self.labels)
indices, scores = indices_scores
offset = 0
for i, doc in enumerate(docs):
@ -200,7 +200,7 @@ class Exclusive_SpanCategorizer(SpanCategorizer):
doc,
indices_i,
scores[offset : offset + indices.lengths[i]],
labels, # type: ignore[arg-type]
labels,
allow_overlap,
)
offset += indices.lengths[i]