Fix tests for history features

This commit is contained in:
Matthew Honnibal 2017-10-09 04:12:24 +02:00
parent be4f0b6460
commit 6c79841c0d
1 changed files with 3 additions and 2 deletions

View File

@ -35,7 +35,8 @@ def parser(vocab, arc_eager):
@pytest.fixture @pytest.fixture
def model(arc_eager, tok2vec): 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 @pytest.fixture
def doc(vocab): def doc(vocab):
@ -51,7 +52,7 @@ def test_can_init_nn_parser(parser):
def test_build_model(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 assert parser.model is not None