Limit parser model size, to hopefully reduce memory during CI tests
@@ -17,7 +17,7 @@ def test_beam_parse():
nlp = Language()
nlp.add_pipe(DependencyParser(nlp.vocab), name='parser')
nlp.parser.add_label('nsubj')
- nlp.begin_training()
+ nlp.parser.begin_training([], token_vector_width=8, hidden_width=8)
doc = nlp.make_doc(u'Australia is a country')
nlp.parser(doc, beam_width=2)
