Add wsd requirements

PyThaiNLP · Jul 12, 2023 · 2216899 · 2216899
1 parent 3e9b28b
commit 2216899
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/docker_requirements.txt b/docker_requirements.txt
@@ -37,3 +37,4 @@ ufal.chu-liu-edmonds==1.0.2
 wtpsplit==1.0.1
 fastcoref==2.1.6
 panphon==0.20.0
+sentence-transformers==2.2.2
diff --git a/docs/notes/installation.rst b/docs/notes/installation.rst
@@ -36,6 +36,7 @@ where ``extras`` can be
   - ``transformers_ud`` (to support transformers_ud engine)
   - ``dependency_parsing`` (to support dependency parsing with all engine)
   - ``coreference_resolution`` (to support coreference esolution with all engine)
+  - ``wsd`` (to support pythainlp.wsd)
   - ``full`` (install everything)
 
 For dependency details, look at `extras` variable in `setup.py <https://github.com/PyThaiNLP/pythainlp/blob/dev/setup.py>`_.

diff --git a/setup.py b/setup.py
@@ -117,6 +117,9 @@
     "word_approximation":{
         "panphon>=0.20.0"
     },
+    "wsd":{
+        "sentence-transformers>=2.2.2"
+    },
     "full": [
         "PyYAML>=5.3.1",
         "attacut>=1.0.4",
@@ -150,6 +153,7 @@
         "fastcoref>=2.1.5",
         "ufal.chu-liu-edmonds>=1.0.2",
         "panphon>=0.20.0",
+        "sentence-transformers>=2.2.2",
     ],
 }