Update pos_tag_transformers()

argument `words` -> `sentence`
PyThaiNLP · Dec 11, 2023 · 4b80066 · 4b80066
1 parent 78ddc17
commit 4b80066
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_tag.py b/tests/test_tag.py
@@ -365,30 +365,32 @@ def test_NNER_class(self):
     def test_pos_tag_transformers(self):
         self.assertIsNotNone(
             pos_tag_transformers(
-                words="แมวทำอะไรตอนห้าโมงเช้า",
+                sentence="แมวทำอะไรตอนห้าโมงเช้า",
                 engine="bert",
                 corpus="blackboard",
             )
         )
         self.assertIsNotNone(
             pos_tag_transformers(
-                words="แมวทำอะไรตอนห้าโมงเช้า", engine="mdeberta", corpus="pud"
+                sentence="แมวทำอะไรตอนห้าโมงเช้า",
+                engine="mdeberta",
+                corpus="pud"
             )
         )
         self.assertIsNotNone(
             pos_tag_transformers(
-                words="แมวทำอะไรตอนห้าโมงเช้า",
+                sentence="แมวทำอะไรตอนห้าโมงเช้า",
                 engine="wangchanberta",
                 corpus="pud",
             )
         )
         with self.assertRaises(ValueError):
             pos_tag_transformers(
-                words="แมวทำอะไรตอนห้าโมงเช้า", engine="non-existing-engine"
+                sentence="แมวทำอะไรตอนห้าโมงเช้า", engine="non-existing-engine"
             )
         with self.assertRaises(ValueError):
             pos_tag_transformers(
-                words="แมวทำอะไรตอนห้าโมงเช้า",
+                sentence="แมวทำอะไรตอนห้าโมงเช้า",
                 engine="bert",
                 corpus="non-existing corpus",
             )