From e0860bcfb3b3d3d71ebf11294729bd79091e1c69 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Fri, 29 Jun 2018 13:56:29 +0200 Subject: [PATCH] Fix bug when docs are empty --- spacy/pipeline.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index d43e6990c..ed4e4c066 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -449,7 +449,7 @@ class Tagger(Pipe): def predict(self, docs): if not any(len(doc) for doc in docs): # Handle case where there are no tokens in any docs. - return [self.model.ops.allocate((0, self.model.nO) for doc in docs] + return [self.model.ops.allocate((0, self.model.nO)) for doc in docs] tokvecs = self.model.tok2vec(docs) scores = self.model.softmax(tokvecs) guesses = []