From 5e16490d9d2cf3fc8c571d56bdbbd36dd11dd759 Mon Sep 17 00:00:00 2001 From: Ines Montani Date: Tue, 5 Feb 2019 12:33:47 +0100 Subject: [PATCH] Fix default argument in TextCategorizer.Model (resolves #3221) --- spacy/pipeline.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index 7a823f1a7..45cba61d2 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -1145,7 +1145,7 @@ class TextCategorizer(Pipe): name = 'textcat' @classmethod - def Model(cls, nr_class, **cfg): + def Model(cls, nr_class=1, **cfg): embed_size = util.env_opt("embed_size", 2000) if "token_vector_width" in cfg: token_vector_width = cfg["token_vector_width"]