Update code

Author: wannaphong

docker_requirements.txt

@@ -33,3 +33,4 @@ wunsen==0.0.3
 khanaa==0.0.6
 spacy_thai==0.7.1
 esupar==1.3.8
+ufal.chu-liu-edmonds==1.0.2

docs/notes/installation.rst

@@ -34,6 +34,7 @@ where ``extras`` can be
   - ``nlpo3`` (to support nlpo3 engine)
   - ``spacy_thai`` (to support spacy_thai engine)
   - ``esupar`` (to support esupar engine)
+  - ``transformers_ud`` (to support transformers_ud engine)
   - ``dependency_parsing`` (to support dependency parsing with all engine)
   - ``full`` (install everything)
 

pythainlp/parse/transformers_ud.py

@@ -39,12 +39,12 @@ def __init__(self, model: str="KoichiYasuoka/deberta-base-thai-ud-head") -> None
             s=AutoConfig.from_pretrained(cached_file(model,"tagger/config.json"))
             t=x(cached_file(model,"tagger/pytorch_model.bin"),config=s)
         self.deprel=TokenClassificationPipeline(
-            model=d
+            model=d,
             tokenizer=self.tokenizer,
             aggregation_strategy="simple"
         )
         self.tagger=TokenClassificationPipeline(
-            model=t
+            model=t,
             tokenizer=self.tokenizer
         )
 

setup.py

@@ -83,9 +83,13 @@
     "thai_nner": ["thai_nner"],
     "esupar": ["esupar>=1.3.8", "numpy"],
     "spacy_thai": ["spacy_thai>=0.7.1"],
+    "transformers_ud": [
+        "ufal.chu-liu-edmonds>=1.0.2"
+    ],
     "dependency_parsing": [
         "esupar>=1.3.8",
         "spacy_thai>=0.7.1",
+        "ufal.chu-liu-edmonds>=1.0.2"
     ],
     "full": [
         "PyYAML>=5.3.1",
@@ -117,6 +121,7 @@
         "wunsen>=0.0.3",
         "spacy_thai>=0.7.1",
         "esupar>=1.3.8",
+        "ufal.chu-liu-edmonds>=1.0.2",
     ],
 }