Change to import inside if block

text/__init__.py

@@ -27,11 +27,6 @@ def get_bert(
     assist_text_weight=0.7,
     ignore_unknown=False,
 ):
-    # from .chinese_bert import get_bert_feature as zh_bert
-    # from .english_bert_mock import get_bert_feature as en_bert
-    # from .japanese_bert import get_bert_feature as jp_bert
-
-    # Change logic to import inside if condition to avoid unnecessary import
     if language == "ZH":
         from .chinese_bert import get_bert_feature as zh_bert
 

text/cleaner.py

@@ -1,33 +1,26 @@
-from text import chinese, japanese, english, cleaned_text_to_sequence
-
-
-language_module_map = {"ZH": chinese, "JP": japanese, "EN": english}
-
-
 def clean_text(text, language, use_jp_extra=True, ignore_unknown=False):
-    language_module = language_module_map[language]
+    # Changed to import inside if condition to avoid unnecessary import
+    if language == "ZH":
+        from . import chinese as language_module
+
+        norm_text = language_module.text_normalize(text)
+        phones, tones, word2ph = language_module.g2p(norm_text)
+    elif language == "EN":
+        from . import english as language_module
+
+        norm_text = language_module.text_normalize(text)
+        phones, tones, word2ph = language_module.g2p(norm_text)
+    elif language == "JP":
+        from . import japanese as language_module
+
         norm_text = language_module.text_normalize(text)
-    if language == "JP":
         phones, tones, word2ph = language_module.g2p(
             norm_text, use_jp_extra, ignore_unknown=ignore_unknown
         )
     else:
-        phones, tones, word2ph = language_module.g2p(norm_text)
+        raise ValueError(f"Language {language} not supported")
     return norm_text, phones, tones, word2ph
 
 
-def clean_text_bert(text, language):
-    language_module = language_module_map[language]
-    norm_text = language_module.text_normalize(text)
-    phones, tones, word2ph = language_module.g2p(norm_text)
-    bert = language_module.get_bert_feature(norm_text, word2ph)
-    return phones, tones, bert
-
-
-def text_to_sequence(text, language):
-    norm_text, phones, tones, word2ph = clean_text(text, language)
-    return cleaned_text_to_sequence(phones, tones, language)
-
-
 if __name__ == "__main__":
     pass