From 4b80066595a77c87dde8346ea1a9892ed8aaf13b Mon Sep 17 00:00:00 2001 From: Arthit Suriyawongkul Date: Mon, 11 Dec 2023 04:05:02 +0000 Subject: [PATCH] Update pos_tag_transformers() argument `words` -> `sentence` --- tests/test_tag.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tests/test_tag.py b/tests/test_tag.py index 5ea859667..9b402980a 100644 --- 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", )