                  self.moves.set_valid(eg.c.is_valid, stcls)
                  fill_context(eg.c.atoms, stcls)
                  self.model.set_scores(eg.c.scores, eg.c.atoms)
                  eg.c.guess = arg_max_if_true(eg.c.scores, eg.c.is_valid, self.model.n_classes)
                  self.moves.c[eg.c.guess].do(stcls, self.moves.c[eg.c.guess].label)
          if stcls.is_final():
              self.moves.finalize_state(stcls)
          tokens.set_parse(stcls._sent)
          return stcls
  
 
</s> remove         cdef Example eg
 </s> add         cdef Example eg = Example(self.model.n_classes, CONTEXT_SIZE,
                                  self.model.n_feats, self.model.n_feats) </s> remove     def get_state(self, Doc tokens, initial_actions, continue_for=0):
 </s> add     def get_state(self, Doc tokens, initial_actions): </s> remove             try:
                action = self.moves.lookup_transition(action_name)
            except IndexError:
                break
            action.do(stcls, action.label)
        else:
            eg = Example(self.model.n_classes, CONTEXT_SIZE,
                         self.model.n_feats, self.model.n_feats)
            while not stcls.is_final() and continue_for != 0:
 </s> add             if action_name == '_':
                action = self.moves.c[eg.c.guess]
            else:
                action = self.moves.lookup_transition(action_name)
            action.do(stcls, action.label)
                  self.moves.set_valid(eg.c.is_valid, stcls)
                  fill_context(eg.c.atoms, stcls)
                  self.model.set_scores(eg.c.scores, eg.c.atoms)
                  eg.c.guess = arg_max_if_true(eg.c.scores, eg.c.is_valid, self.model.n_classes)
                action = self.moves.c[eg.c.guess]
            else:
                action = self.moves.lookup_transition(action_name)
            action.do(stcls, action.label)
          if stcls.is_final():
              self.moves.finalize_state(stcls)
          tokens.set_parse(stcls._sent)
          return stcls
  
keep keep keep keep replace keep keep keep keep keep
 * Fix get_state for parser prediction 
spacy/syntax/parser.pyx