explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-04 21:50:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add type for indices

Browse Source
This commit is contained in:
Adriane Boyd 2023-04-27 09:27:31 +02:00 committed by GitHub
parent eb8d45291c
commit b2db85e209
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/ml/extract_spans.py
View File

@ -1,4 +1,4 @@
from typing import Tuple, Callable from typing import List, Tuple, Callable
from thinc.api import Model, to_numpy from thinc.api import Model, to_numpy
from thinc.types import Ragged, Ints1d from thinc.types import Ragged, Ints1d
@ -52,7 +52,7 @@ def _get_span_indices(ops, spans: Ragged, lengths: Ints1d) -> Ints1d:
indices will be [5, 6, 7, 8, 8, 9]. indices will be [5, 6, 7, 8, 8, 9].
""" """
spans, lengths = _ensure_cpu(spans, lengths) spans, lengths = _ensure_cpu(spans, lengths)
indices = [] indices: List[int] = []
offset = 0 offset = 0
for i, length in enumerate(lengths): for i, length in enumerate(lengths):
spans_i = spans[i].dataXd + offset spans_i = spans[i].dataXd + offset