From 2e2f8785209876e6a3dbb666567e6a2c2109d4c6 Mon Sep 17 00:00:00 2001 From: Wannaphong Phatthiyaphaibun Date: Tue, 12 Dec 2023 20:50:13 +0700 Subject: [PATCH] Update common.py --- pythainlp/corpus/common.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pythainlp/corpus/common.py b/pythainlp/corpus/common.py index 5bf9ac4d7..a2183397f 100644 --- a/pythainlp/corpus/common.py +++ b/pythainlp/corpus/common.py @@ -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)))