diff --git a/spacy/syntax/arc_eager.pyx b/spacy/syntax/arc_eager.pyx index 3694ddc24..30314a227 100644 --- a/spacy/syntax/arc_eager.pyx +++ b/spacy/syntax/arc_eager.pyx @@ -110,7 +110,8 @@ cdef bint _is_gold_root(const GoldParseC* gold, int word) nogil: cdef class Shift: @staticmethod cdef bint is_valid(const StateC* st, attr_t label) nogil: - return st.buffer_length() >= 2 and not st.shifted[st.B(0)] and st.B_(0).sent_start != 1 + sent_start = st._sent[st.B_(0).l_edge].sent_start + return st.buffer_length() >= 2 and not st.shifted[st.B(0)] and sent_start != 1 @staticmethod cdef int transition(StateC* st, attr_t label) nogil: @@ -170,7 +171,8 @@ cdef class Reduce: cdef class LeftArc: @staticmethod cdef bint is_valid(const StateC* st, attr_t label) nogil: - return st.B_(0).sent_start != 1 + sent_start = st._sent[st.B_(0).l_edge].sent_start + return sent_start != 1 @staticmethod cdef int transition(StateC* st, attr_t label) nogil: @@ -205,7 +207,8 @@ cdef class RightArc: @staticmethod cdef bint is_valid(const StateC* st, attr_t label) nogil: # If there's (perhaps partial) parse pre-set, don't allow cycle. - return st.B_(0).sent_start != 1 and st.H(st.S(0)) != st.B(0) + sent_start = st._sent[st.B_(0).l_edge].sent_start + return sent_start != 1 and st.H(st.S(0)) != st.B(0) @staticmethod cdef int transition(StateC* st, attr_t label) nogil: