From b9df54adcd496fc2f618f5f3c0d3d25519bdfcfb Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 17 Mar 2020 15:05:13 +0100 Subject: [PATCH] bug fix and check cleaner config field by comparing with the list of avail functions --- utils/generic_utils.py | 5 +++-- utils/text/cleaners.py | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/generic_utils.py b/utils/generic_utils.py index f6c38530..9af0bb70 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -496,7 +496,8 @@ def check_config(c): _check_argument('tb_model_param_stats', c, restricted=True, val_type=bool) # dataloading - _check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=['english_cleaners', 'phoneme_cleaners', 'transliteration_cleaners', 'basic_cleaners']) + from TTS.utils.text import cleaners + _check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=dir(cleaners)) _check_argument('enable_eos_bos_chars', c, restricted=True, val_type=bool) _check_argument('num_loader_workers', c, restricted=True, val_type=int, min_val=0) _check_argument('num_val_loader_workers', c, restricted=True, val_type=int, min_val=0) @@ -518,4 +519,4 @@ def check_config(c): _check_argument('name', dataset_entry, restricted=True, val_type=str) _check_argument('path', dataset_entry, restricted=True, val_type=str) _check_argument('meta_file_train', dataset_entry, restricted=True, val_type=str) - _check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str) \ No newline at end of file + _check_argument('meta_file_val', dataset_entry, restricted=True, val_type=str) diff --git a/utils/text/cleaners.py b/utils/text/cleaners.py index 92c2d934..35da8aef 100644 --- a/utils/text/cleaners.py +++ b/utils/text/cleaners.py @@ -97,6 +97,7 @@ def basic_turkish_cleaners(text): text = text.replace("I", "ı") text = lowercase(text) text = collapse_whitespace(text) + return text def english_cleaners(text):