diff --git a/spacy_lookups_data/tests/test_lt.py b/spacy_lookups_data/tests/test_lt.py index a46d34b..1aa1be6 100644 --- a/spacy_lookups_data/tests/test_lt.py +++ b/spacy_lookups_data/tests/test_lt.py @@ -25,4 +25,4 @@ def lt_lemmatizer(): @pytest.mark.parametrize("tokens,lemmas", TEST_CASES) def test_lt_lemmatizer(lt_lemmatizer, tokens, lemmas): - assert lemmas == [lt_lemmatizer.lookup_table.get(token, token) for token in tokens] + assert lemmas == [lt_lemmatizer.lookup(token) for token in tokens]