Extend to stanza v1.4

explosion · May 27, 2022 · 5895c99 · 5895c99
1 parent 7a9faac
commit 5895c99
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 spacy>=3.0.0,<4.0.0
-stanza>=1.2.0,<1.4.0
+stanza>=1.2.0,<1.5.0
 # Development dependencies
 pytest>=5.2.0
diff --git a/setup.py b/setup.py
@@ -32,7 +32,7 @@ def setup_package():
         version=about["__version__"],
         license=about["__license__"],
         packages=find_packages(),
-        install_requires=["spacy>=3.0.0,<4.0.0", "stanza>=1.2.0,<1.4.0"],
+        install_requires=["spacy>=3.0.0,<4.0.0", "stanza>=1.2.0,<1.5.0"],
         python_requires=">=3.6",
         entry_points={
             "spacy_tokenizers": [

diff --git a/tests/test_language.py b/tests/test_language.py
@@ -31,7 +31,7 @@ def test_spacy_stanza_english():
 
     assert [t.tag_ for t in doc] == ["UH", "NN", ".", "DT", "VBZ", "DT", "NN", '.']
     assert [str(t.morph) for t in doc] == ['', 'Number=Sing', '', 'Number=Sing|PronType=Dem', 'Mood=Ind|Number=Sing|Person=3|Tense=Pres|VerbForm=Fin', 'Definite=Ind|PronType=Art', 'Number=Sing', '']
-    assert [t.dep_ for t in doc] == ['discourse', 'root', 'punct', 'nsubj', 'cop', 'det', 'root', 'punct']
+    assert [t.dep_ for t in doc] == ['root', 'vocative', 'punct', 'nsubj', 'cop', 'det', 'root', 'punct']
     assert [t.is_sent_start for t in doc] == [True, False, False, True, False, False, False, False]
     assert any([t.is_stop for t in doc])
     # fmt: on