Fix tagger when doc is empty
@@ -450,7 +450,9 @@ class Tagger(Pipe):
if not any(len(doc) for doc in docs):
# Handle case where there are no tokens in any docs.
n_labels = len(self.labels)
- return [self.model.ops.allocate((0, n_labels)) for doc in docs]
+ guesses = [self.model.ops.allocate((0, n_labels)) for doc in docs]
+ tokvecs = self.model.ops.allocate((0, self.model.tok2vec.nO))
+ return guesses, tokvecs
tokvecs = self.model.tok2vec(docs)
scores = self.model.softmax(tokvecs)
guesses = []
