mirror of https://github.com/explosion/spaCy.git
Fix tests for history features
This commit is contained in:
parent
be4f0b6460
commit
6c79841c0d
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue