explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-27 17:12:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update test

Browse Source
This commit is contained in:
Matthew Honnibal 2018-05-07 16:59:52 +02:00
parent f6cdafc00e
commit 01c4e13b02
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/tests/parser/test_neural_parser.py
View File

@ -35,8 +35,7 @@ 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, return Parser.Model(arc_eager.n_moves, token_vector_width=tok2vec.nO)[0]
hist_size=0)[0]
@pytest.fixture @pytest.fixture
def doc(vocab): def doc(vocab):
@ -69,11 +68,13 @@ def test_update_doc(parser, model, doc, gold):
parser.update([doc], [gold], sgd=optimize) parser.update([doc], [gold], sgd=optimize)
@pytest.mark.xfail
def test_predict_doc_beam(parser, model, doc): def test_predict_doc_beam(parser, model, doc):
parser.model = model parser.model = model
parser(doc, beam_width=32, beam_density=0.001) parser(doc, beam_width=32, beam_density=0.001)
@pytest.mark.xfail
def test_update_doc_beam(parser, model, doc, gold): def test_update_doc_beam(parser, model, doc, gold):
parser.model = model parser.model = model
def optimize(weights, gradient, key=None): def optimize(weights, gradient, key=None):