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

Undo changes to extract_spans

Browse Source
This commit is contained in:
Adriane Boyd 2022-11-29 13:17:28 +01:00
parent df73823aae
commit 4ecccf8460
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

@ -32,8 +32,8 @@ def forward(
Y = Ragged(X.dataXd[indices], spans.dataXd[:, 1] - spans.dataXd[:, 0]) # type: ignore[arg-type, index]
else:
Y = Ragged(
ops.xp.zeros((0, 0), dtype=X.dataXd.dtype),
ops.xp.zeros((0,), dtype="i"),
ops.xp.zeros(X.dataXd.shape, dtype=X.dataXd.dtype),
ops.xp.zeros((len(X.lengths),), dtype="i"),
)
x_shape = X.dataXd.shape
x_lengths = X.lengths