def test_issue7056(): """Test that the Unshift transition works properly, and doesn't cause sentence segmentation errors.""" vocab = Vocab() ae = ArcEager( vocab.strings, ArcEager.get_actions(left_labels=["amod"], right_labels=["pobj"])) doc = Doc(vocab, words="Severe pain , after trauma".split()) state = ae.init_batch([doc])[0] ae.apply_transition(state, "S") ae.apply_transition(state, "L-amod") ae.apply_transition(state, "S") ae.apply_transition(state, "S") ae.apply_transition(state, "S") ae.apply_transition(state, "R-pobj") ae.apply_transition(state, "D") ae.apply_transition(state, "D") ae.apply_transition(state, "D") assert not state.eol()
def arc_eager(vocab): actions = ArcEager.get_actions(left_labels=["L"], right_labels=["R"]) return ArcEager(vocab.strings, actions)
def arc_eager(vocab): moves = ArcEager(vocab.strings, ArcEager.get_actions()) moves.add_action(2, "left") moves.add_action(3, "right") return moves
def moves(vocab): aeager = ArcEager(vocab.strings, {}) aeager.add_action(0, "") aeager.add_action(1, "") aeager.add_action(2, "nsubj") aeager.add_action(2, "punct") aeager.add_action(2, "aux") aeager.add_action(2, "nsubjpass") aeager.add_action(3, "dobj") aeager.add_action(2, "aux") aeager.add_action(4, "ROOT") return aeager