From d85d063231850dc0e56e1de82776c56b4d2d314e Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 23 Jun 2020 23:58:55 +0200 Subject: [PATCH] Fix NER init_gold_batch --- spacy/syntax/ner.pyx | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spacy/syntax/ner.pyx b/spacy/syntax/ner.pyx index 13e884c0c..3fc64cb82 100644 --- a/spacy/syntax/ner.pyx +++ b/spacy/syntax/ner.pyx @@ -133,9 +133,8 @@ cdef class BiluoPushDown(TransitionSystem): examples = [eg for eg in examples if self.has_gold(eg)] states = self.init_batch([eg.predicted for eg in examples]) keeps = [i for i, s in enumerate(states) if not s.is_final()] - states = [states[i] for i in keeps] - examples = [examples[i] for i in keeps] golds = [BiluoGold(self, states[i], examples[i]) for i in keeps] + states = [states[i] for i in keeps] n_steps = sum([len(s.queue) for s in states]) return states, golds, n_steps