explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

spancat type fixes

Browse Source
This commit is contained in:
Daniël de Kok 2023-06-12 16:43:05 +02:00
parent d82e167aea
commit 4990cfefb4
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/pipeline/spancat.py
View File

@ -537,10 +537,11 @@ class SpanCategorizer(TrainablePipe):
DOCS: https://spacy.io/api/spancategorizer#set_annotations
"""
indices = activations["indices"]
scores = activations["scores"]
assert isinstance(indices, Ragged)
scores = cast(Floats2d, activations["scores"])
offset = 0
for i, doc in enumerate(docs):
indices_i = indices[i].dataXd
indices_i = cast(Ints2d, indices[i].dataXd)
if self.save_activations:
doc.activations[self.name] = {}
doc.activations[self.name]["indices"] = indices_i