explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-11 04:08:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix ArcEager.set_valid

Browse Source
This commit is contained in:
Matthew Honnibal 2016-02-01 03:07:37 +01:00
parent 9c34ca9e5d
commit cb78d91ec5
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/syntax/arc_eager.pyx
View File

@ -395,13 +395,13 @@ cdef class ArcEager(TransitionSystem):
elif USE_ROOT_ARC_SEGMENT and st.c._sent[i].dep == self.root_label:
st.c._sent[i].head = 0
cdef int set_valid(self, int* output, StateClass stcls) nogil:
cdef int set_valid(self, int* output, const StateC* st) nogil:
cdef bint[N_MOVES] is_valid
is_valid[SHIFT] = Shift.is_valid(stcls.c, -1)
is_valid[REDUCE] = Reduce.is_valid(stcls.c, -1)
is_valid[LEFT] = LeftArc.is_valid(stcls.c, -1)
is_valid[RIGHT] = RightArc.is_valid(stcls.c, -1)
is_valid[BREAK] = Break.is_valid(stcls.c, -1)
is_valid[SHIFT] = Shift.is_valid(st, -1)
is_valid[REDUCE] = Reduce.is_valid(st, -1)
is_valid[LEFT] = LeftArc.is_valid(st, -1)
is_valid[RIGHT] = RightArc.is_valid(st, -1)
is_valid[BREAK] = Break.is_valid(st, -1)
cdef int i
for i in range(self.n_moves):
output[i] = is_valid[self.c[i].move]