From 0ce6cde5877633a3243a09db4861adf7b03cda0b Mon Sep 17 00:00:00 2001 From: Pavarissy Date: Sun, 6 Oct 2024 14:28:05 +0000 Subject: [PATCH 1/2] add preprocess function for romanizing --- pythainlp/transliterate/core.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pythainlp/transliterate/core.py b/pythainlp/transliterate/core.py index a7019393..1124b78a 100644 --- a/pythainlp/transliterate/core.py +++ b/pythainlp/transliterate/core.py @@ -84,7 +84,12 @@ def select_romanize_engine(engine: str): fallback = select_romanize_engine(fallback_engine) return romanize(text, fallback_func=fallback) else: - return select_romanize_engine(engine)(text) + rom_engine = select_romanize_engine(engine) + trans_word=[] + for word in text.split(' '): + trans_word.append(rom_engine(word)) + new_word=''.join(trans_word) + return new_word def transliterate( From 0910d727e6b43b5c0c5401225bc6ab6940599155 Mon Sep 17 00:00:00 2001 From: Pavarissy Date: Sun, 6 Oct 2024 14:55:29 +0000 Subject: [PATCH 2/2] fix pep8 --- pythainlp/transliterate/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pythainlp/transliterate/core.py b/pythainlp/transliterate/core.py index 1124b78a..b8fc1584 100644 --- a/pythainlp/transliterate/core.py +++ b/pythainlp/transliterate/core.py @@ -85,10 +85,10 @@ def select_romanize_engine(engine: str): return romanize(text, fallback_func=fallback) else: rom_engine = select_romanize_engine(engine) - trans_word=[] + trans_word = [] for word in text.split(' '): trans_word.append(rom_engine(word)) - new_word=''.join(trans_word) + new_word = ''.join(trans_word) return new_word