explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 09:26:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

remove dtype (#11615)

Browse Source
This commit is contained in:
Sofie Van Landeghem 2022-10-11 15:25:05 +02:00 committed by GitHub
parent ef74f8f5e4
commit 29649589fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/edit_tree_lemmatizer.py
View File

@ -149,7 +149,7 @@ class EditTreeLemmatizer(TrainablePipe):
# Handle cases where there are no tokens in any docs. # Handle cases where there are no tokens in any docs.
n_labels = len(self.cfg["labels"]) n_labels = len(self.cfg["labels"])
guesses: List[Ints2d] = [ guesses: List[Ints2d] = [
self.model.ops.alloc2i(0, n_labels, dtype="i") for _ in docs self.model.ops.alloc2i(0, n_labels) for _ in docs
] ]
assert len(guesses) == n_docs assert len(guesses) == n_docs
return guesses return guesses