From 6c79841c0dce58cff859dfd37c44824cd267d8ea Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 9 Oct 2017 04:12:24 +0200 Subject: [PATCH] Fix tests for history features --- spacy/tests/parser/test_neural_parser.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spacy/tests/parser/test_neural_parser.py b/spacy/tests/parser/test_neural_parser.py index 8747b01ba..ae20cd5f0 100644 --- a/spacy/tests/parser/test_neural_parser.py +++ b/spacy/tests/parser/test_neural_parser.py @@ -35,7 +35,8 @@ def parser(vocab, arc_eager): @pytest.fixture def model(arc_eager, tok2vec): - return Parser.Model(arc_eager.n_moves, token_vector_width=tok2vec.nO)[0] + return Parser.Model(arc_eager.n_moves, token_vector_width=tok2vec.nO, + hist_size=0)[0] @pytest.fixture def doc(vocab): @@ -51,7 +52,7 @@ def test_can_init_nn_parser(parser): def test_build_model(parser): - parser.model = Parser.Model(parser.moves.n_moves)[0] + parser.model = Parser.Model(parser.moves.n_moves, hist_size=0)[0] assert parser.model is not None