fix add_label test

This commit is contained in:
svlandeg 2020-06-22 11:58:59 +02:00
parent 8ba8defa78
commit ffddff03b8

View File

@ -54,7 +54,8 @@ def test_add_label(parser):
losses = {}
doc = Doc(parser.vocab, words=["a", "b", "c", "d"])
gold = {"heads": [1, 1, 3, 3], "deps": ["right", "ROOT", "left", "ROOT"]}
parser.update((doc, gold), sgd=sgd, losses=losses)
example = Example.from_dict(doc, gold)
parser.update([example], sgd=sgd, losses=losses)
doc = Doc(parser.vocab, words=["a", "b", "c", "d"])
doc = parser(doc)
assert doc[0].dep_ == "right"