diff --git a/spacy/syntax/parser.pxd b/spacy/syntax/parser.pxd index e4fc38969..7803530ff 100644 --- a/spacy/syntax/parser.pxd +++ b/spacy/syntax/parser.pxd @@ -1,4 +1,4 @@ -from .._ml cimport Model, HastyModel +from .._ml cimport Model from .arc_eager cimport TransitionSystem diff --git a/spacy/syntax/parser.pyx b/spacy/syntax/parser.pyx index 4139a9652..ab3fd89c5 100644 --- a/spacy/syntax/parser.pyx +++ b/spacy/syntax/parser.pyx @@ -63,7 +63,7 @@ cdef class GreedyParser: self.cfg = Config.read(model_dir, 'config') self.moves = TransitionSystem(self.cfg.left_labels, self.cfg.right_labels) hasty_templ, full_templ = get_templates(self.cfg.features) - self.model = HastyModel(self.moves.n_moves, hasty_templ, full_templ, + self.model = Model(self.moves.n_moves, hasty_templ, full_templ, model_dir) cpdef int parse(self, Tokens tokens) except -1: