From 1d3884c46ddb32ca009baa9d25c4ca05443d3d0d Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sat, 7 Nov 2015 10:03:56 +1100 Subject: [PATCH] * Fix test_basic_create --- spacy/tests/test_basic_create.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spacy/tests/test_basic_create.py b/spacy/tests/test_basic_create.py index 322efee4a..5019196d5 100644 --- a/spacy/tests/test_basic_create.py +++ b/spacy/tests/test_basic_create.py @@ -11,8 +11,8 @@ from spacy.strings import StringStore from spacy.vocab import Vocab from spacy.tokenizer import Tokenizer from spacy.syntax.arc_eager import ArcEager -from spacy.tagger import Tagger -from spacy.syntax.parser import Parser +from spacy.tagger import Tagger, TaggerModel +from spacy.syntax.parser import Parser, ParserModel from spacy.matcher import Matcher @@ -63,7 +63,7 @@ class TestTagger(unittest.TestCase): def test_create(self): vocab = Vocab() templates = ((1,),) - model = Model(vocab.morphology.n_tags, templates, model_loc=None) + model = TaggerModel(vocab.morphology.n_tags, templates, model_loc=None) tagger = Tagger(vocab, model) @@ -73,7 +73,7 @@ class TestParser(unittest.TestCase): templates = ((1,),) labels_by_action = {0: ['One', 'Two'], 1: ['Two', 'Three']} transition_system = ArcEager(vocab.strings, labels_by_action) - model = Model(vocab.morphology.n_tags, templates, model_loc=None) + model = ParserModel(transition_system.n_moves, templates) parser = Parser(vocab.strings, transition_system, model)