Update conllu training script

examples/training/conllu.py

@@ -352,14 +352,15 @@ class TreebankPaths(object):
     config=("Path to json formatted config file", "positional", None, Config.load),
     corpus=("UD corpus to train and evaluate on, e.g. en, es_ancora, etc",
             "positional", None, str),
-    parses=("Path to write the development parses", "positional", None, Path)
+    parses_loc=("Path to write the development parses", "positional", None, Path),
+    limit=("Size limit", "option", "n", int)
 )
-def main(ud_dir, corpus, config, parses='/tmp/dev.conllu'):
+def main(ud_dir, corpus, config, parses_loc='/tmp/dev.conllu', limit=10):
     paths = TreebankPaths(ud_dir, corpus)
     nlp = load_nlp(paths.lang, config)
 
     docs, golds = read_data(nlp, paths.train.conllu.open(), paths.train.text.open(),
-                            config)
+                            limit=limit)
 
     optimizer = initialize_pipeline(nlp, docs, golds, config)
     n_train_words = sum(len(doc) for doc in docs)
@@ -379,7 +380,7 @@ def main(ud_dir, corpus, config, parses='/tmp/dev.conllu'):
         with nlp.use_params(optimizer.averages):
             dev_docs, scorer = parse_dev_data(nlp, paths.dev.text, paths.dev.conllu)
             print_progress(i, losses, scorer)
-            with open(output_loc, 'w') as file_:
+            with open(parses_loc, 'w') as file_:
                 print_conllu(dev_docs, file_)