fixed spacy language map issue (#72)

* fixed spacy language map issue * fixed spacy version * fixed spacy reqs * changed to greater than spacy 3.0.6 in reqs Co-authored-by: McClain Thiel <mthiel@nference.com>
Georgetown-IR-Lab · Jul 2, 2021 · e6cbade · e6cbade
1 parent c0b5db0
commit e6cbade
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/quickumls/constants.py b/quickumls/constants.py
@@ -185,12 +185,12 @@
 }
 
 SPACY_LANGUAGE_MAP = {
-    'ENG': 'en',
-    'GER': 'de',
-    'SPA': 'es',
-    'POR': 'pt',
-    'FRE': 'fr',
-    'ITA': 'it',
-    'DUT': 'nl',
+    'ENG': 'en_core_web_sm',
+    'GER': 'de_core_news_sm',
+    'SPA': 'es_core_news_sm',
+    'POR': 'pt_core_news_sm',
+    'FRE': 'fr_core_news_sm',
+    'ITA': 'it_core_news_sm',
+    'DUT': 'nl_core_news_sm',
     'XXX': 'xx'
 }
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 leveldb>=0.193
 numpy>=1.8.2
-spacy>=1.6.0
+spacy>=3.0.6
 unidecode>=0.4.19
 nltk>=3.3
 quickumls_simstring>=1.1.5r1
-unqlite>=0.8.1
+unqlite>=0.8.1