Skip to content

Commit

Permalink
Update common.py
Browse files Browse the repository at this point in the history
  • Loading branch information
wannaphong committed Dec 12, 2023
1 parent 92ad41a commit 2e2f878
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pythainlp/corpus/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -359,15 +359,15 @@ def find_synonyms(word: str) -> List[str]:
synonyms = thai_synonyms() # get a dictionary of {word, synonym}
list_synonym = []

if word in synonyms["word"]: # find by word
if word in synonyms["word"]: # find by word
list_synonym.extend(synonyms["synonym"][synonyms["word"].index(word)])

for idx, words in enumerate(synonyms["synonym"]): # find by synonym
for idx, words in enumerate(synonyms["synonym"]): # find by synonym
if word in words:
list_synonym.extend(synonyms["synonym"][idx])
list_synonym.append(synonyms["word"][idx])

if word in list_synonym: # remove same word
if word in list_synonym: # remove same word
list_synonym.remove(word)

return sorted(list(set(list_synonym)))

0 comments on commit 2e2f878

Please sign in to comment.