From 83d1a1e51249d80d188af624ab6c3c8f3501e24d Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 8 Sep 2015 15:39:43 +0200 Subject: [PATCH] * Fix lemmatizer tests --- tests/tagger/test_lemmatizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tagger/test_lemmatizer.py b/tests/tagger/test_lemmatizer.py index 5a6a8fc62..8461a854e 100644 --- a/tests/tagger/test_lemmatizer.py +++ b/tests/tagger/test_lemmatizer.py @@ -1,6 +1,6 @@ from __future__ import unicode_literals -from spacy.en.lemmatizer import Lemmatizer, read_index, read_exc +from spacy.lemmatizer import Lemmatizer, read_index, read_exc from spacy.en import LOCAL_DATA_DIR from os import path @@ -23,7 +23,7 @@ def test_read_exc(): @pytest.fixture def lemmatizer(): - return Lemmatizer(path.join(LOCAL_DATA_DIR, 'wordnet'), 0, 0, 0) + return Lemmatizer.from_dir(path.join(LOCAL_DATA_DIR)) def test_noun_lemmas(lemmatizer):