remove dtype (#11615)

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

View File

@ -149,7 +149,7 @@ class EditTreeLemmatizer(TrainablePipe):
# Handle cases where there are no tokens in any docs.
n_labels = len(self.cfg["labels"])
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
return guesses