From 4cd1c743857c082a998ca0eabbf2b100b0f9f8a3 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 22 Jun 2020 16:00:55 +0200 Subject: [PATCH] Update test --- spacy/tests/parser/test_arc_eager_oracle.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spacy/tests/parser/test_arc_eager_oracle.py b/spacy/tests/parser/test_arc_eager_oracle.py index 954fd9a8a..081be6df3 100644 --- a/spacy/tests/parser/test_arc_eager_oracle.py +++ b/spacy/tests/parser/test_arc_eager_oracle.py @@ -66,7 +66,8 @@ def test_oracle_four_words(arc_eager, vocab): heads = [1, 1, 3, 3] deps = ["left", "ROOT", "left", "ROOT"] for dep in deps: - arc_eager.add_label(dep) + arc_eager.add_action(2, dep) # Left + arc_eager.add_action(3, dep) # Right actions = ["L-left", "B-ROOT", "L-left"] state, cost_history = get_sequence_costs(arc_eager, words, heads, deps, actions) assert state.is_final() @@ -143,7 +144,7 @@ def test_get_oracle_actions(): doc = Doc(Vocab(), words=[t[1] for t in annot_tuples]) config = { "learn_tokens": False, - "min_action_freq": 30, + "min_action_freq": 0, "beam_width": 1, "beam_update_prob": 1.0, }