Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

* Fix bounds checking on entities

This commit is contained in:
Matthew Honnibal 2015-06-23 04:35:08 +02:00
parent 43ef5ddea5
commit ee3e56f27b
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
spacy/syntax/stateclass.pyx
View File

@ -35,9 +35,9 @@ cdef class StateClass:
cdef int E(self, int i) nogil: cdef int E(self, int i) nogil:
if self._e_i <= 0 or self._e_i >= self.length: if self._e_i <= 0 or self._e_i >= self.length:
return -1 return 0
if i <= 0 or i >= self.length: if i < 0 or i >= self.length:
return -1 return 0
return self._ents[self._e_i-1].start return self._ents[self._e_i-1].start
cdef int L(self, int i, int idx) nogil: cdef int L(self, int i, int idx) nogil:
@ -203,11 +203,10 @@ cdef class StateClass:
self._sent[head].l_kids -= 1 self._sent[head].l_kids -= 1
cdef void open_ent(self, int label) nogil: cdef void open_ent(self, int label) nogil:
if 0 <= self._e_i < self.length: self._ents[self._e_i].start = self.B(0)
self._ents[self._e_i].start = self.B(0) self._ents[self._e_i].label = label
self._ents[self._e_i].label = label self._ents[self._e_i].end = -1
self._ents[self._e_i].end = -1 self._e_i += 1
self._e_i += 1
cdef void close_ent(self) nogil: cdef void close_ent(self) nogil:
self._ents[self._e_i-1].end = self.B(0)+1 self._ents[self._e_i-1].end = self.B(0)+1