From 2216899854e5f9b55f5c393d9de1ed7ec8d81864 Mon Sep 17 00:00:00 2001
From: Wannaphong Phatthiyaphaibun <wannaphong@yahoo.com>
Date: Wed, 12 Jul 2023 01:39:10 -0700
Subject: [PATCH] Add wsd requirements

---
 docker_requirements.txt     | 1 +
 docs/notes/installation.rst | 1 +
 setup.py                    | 4 ++++
 3 files changed, 6 insertions(+)

diff --git a/docker_requirements.txt b/docker_requirements.txt
index 4cd8b63f9..c5fe9591e 100644
--- 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
index fa5bdb896..92bbd436e 100644
--- 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
index 7da84e696..16bb96c1a 100644
--- 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",
     ],
 }