remove dtype
@@ -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
