Limit parser model size, to hopefully reduce memory during CI tests

2018-01-28 21:00:32 +01:00 · 2018-01-28 21:00:32 +01:00 · f5b1ad4100
parent 6d978e5c35
commit f5b1ad4100
1 changed files with 1 additions and 1 deletions
--- a/spacy/tests/parser/test_beam_parse.py
+++ b/spacy/tests/parser/test_beam_parse.py
@ -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)