Fix g2pw bug (#1872)

This commit is contained in:
wjd 2024-12-19 14:30:57 +08:00 committed by GitHub
parent a70e1ad30c
commit 793b0043de
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -144,11 +144,16 @@ def read_dict():
return polyphonic_dict
def correct_pronunciation(word,word_pinyins):
if word in pp_dict:
word_pinyins = pp_dict[word]
def correct_pronunciation(word, word_pinyins):
new_pinyins = pp_dict.get(word, "")
if new_pinyins == "":
for idx, w in enumerate(word):
w_pinyin = pp_dict.get(w, "")
if w_pinyin != "":
word_pinyins[idx] = w_pinyin[0]
return word_pinyins
else:
return new_pinyins
pp_dict = get_dict()