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

bugfix: iterators now reset properly when called a second time

This commit is contained in:
Wolfgang Seeker 2016-04-15 17:49:16 +02:00
parent e6945c4d0e
commit b98cc3266d
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/syntax/iterators.pyx
View File

@ -28,6 +28,10 @@ cdef class EnglishNounChunks(DocIterator):
self._conjunct = self._doc.vocab.strings['conj'] self._conjunct = self._doc.vocab.strings['conj']
self.i = 0 self.i = 0
def __iter__(self):
self.i = 0
return super(EnglishNounChunks,self).__iter__()
def __next__(self): def __next__(self):
cdef const TokenC* word cdef const TokenC* word
cdef widx cdef widx
@ -63,6 +67,10 @@ cdef class GermanNounChunks(DocIterator):
self._close_app = self._doc.vocab.strings['nk'] self._close_app = self._doc.vocab.strings['nk']
self.i = 0 self.i = 0
def __iter__(self):
self.i = 0
return super(GermanNounChunks,self).__iter__()
def __next__(self): def __next__(self):
cdef const TokenC* word cdef const TokenC* word
cdef int rbracket cdef int rbracket
@ -79,6 +87,6 @@ cdef class GermanNounChunks(DocIterator):
for rdep in self._doc[widx].rights: for rdep in self._doc[widx].rights:
if rdep.pos == NOUN and rdep.dep == self._close_app: if rdep.pos == NOUN and rdep.dep == self._close_app:
rbracket = rdep.i+1 rbracket = rdep.i+1
return Span(self._doc, word.l_edge, rbracket, label=self._np_label) return Span(self._doc, word.l_edge, rbracket, label=self._np_label)
raise StopIteration raise StopIteration