Update pronounce.py

PyThaiNLP · Oct 18, 2023 · 1cd22ba · 1cd22ba
1 parent 7ddeb04
commit 1cd22ba
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pythainlp/util/pronounce.py b/pythainlp/util/pronounce.py
@@ -20,10 +20,12 @@
 
 
 kv = KhaveeVerifier()
-all_thai_words_dict = [i for i in list(thai_words()) if len(syllable_tokenize(i))==1]
+all_thai_words_dict = [
+    i for i in list(thai_words()) if len(syllable_tokenize(i)) == 1
+]
 
 
-def rhyme(word: str)->List[str]:
+def rhyme(word: str) -> List[str]:
     """
     Find Thai rhyme
 
@@ -38,8 +40,8 @@ def rhyme(word: str)->List[str]:
         print(rhyme("จีบ"))
         # output: ['กลีบ', 'กีบ', 'ครีบ', ...]
     """
-    list_sumpus=[]
+    list_sumpus = []
     for i in all_thai_words_dict:
-        if kv.is_sumpus(word,i) and i!=word:
+        if kv.is_sumpus(word, i) and i != word:
             list_sumpus.append(i)
     return sorted(list_sumpus)