From ae8cf70dc1704c38899d8ca488f0bbde992b4834 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 22 May 2017 06:13:39 -0500 Subject: [PATCH] Fix CLI train signature --- spacy/cli/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/cli/train.py b/spacy/cli/train.py index a9a5cd536..99d05747d 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -31,8 +31,8 @@ from .. import displacy no_parser=("Don't train parser", "flag", "P", bool), no_entities=("Don't train NER", "flag", "N", bool) ) -def train(lang, output_dir, train_data, dev_data, n_iter, n_sents, - use_gpu, no_tagger, no_parser, no_entities): +def train(_, lang, output_dir, train_data, dev_data, n_iter=20, n_sents=0, + use_gpu=False, no_tagger=False, no_parser=False, no_entities=False): """Train a model. Expects data in spaCy's JSON format.""" n_sents = n_sents or None output_path = util.ensure_path(output_dir)