mirror of https://github.com/coqui-ai/TTS.git
Fix tests
This commit is contained in:
parent
21940952bf
commit
bc2243bac4
|
@ -7,14 +7,16 @@ from tqdm.contrib.concurrent import process_map
|
||||||
|
|
||||||
from TTS.config import load_config
|
from TTS.config import load_config
|
||||||
from TTS.tts.datasets import load_tts_samples
|
from TTS.tts.datasets import load_tts_samples
|
||||||
from TTS.tts.utils.text import text2phone
|
from TTS.tts.utils.text.phonemizers.gruut_wrapper import Gruut
|
||||||
|
|
||||||
|
|
||||||
|
phonemizer = Gruut(language="en-us")
|
||||||
|
|
||||||
|
|
||||||
def compute_phonemes(item):
|
def compute_phonemes(item):
|
||||||
try:
|
try:
|
||||||
text = item[0]
|
text = item[0]
|
||||||
language = item[-1]
|
ph = phonemizer.phonemize(text).split("|")
|
||||||
ph = text2phone(text, language, use_espeak_phonemes=c.use_espeak_phonemes).split("|")
|
|
||||||
except:
|
except:
|
||||||
return []
|
return []
|
||||||
return list(set(ph))
|
return list(set(ph))
|
||||||
|
|
|
@ -39,7 +39,6 @@ class TestFindUniquePhonemes(unittest.TestCase):
|
||||||
num_eval_loader_workers=0,
|
num_eval_loader_workers=0,
|
||||||
text_cleaner="english_cleaners",
|
text_cleaner="english_cleaners",
|
||||||
use_phonemes=True,
|
use_phonemes=True,
|
||||||
use_espeak_phonemes=True,
|
|
||||||
phoneme_language="en-us",
|
phoneme_language="en-us",
|
||||||
phoneme_cache_path="tests/data/ljspeech/phoneme_cache/",
|
phoneme_cache_path="tests/data/ljspeech/phoneme_cache/",
|
||||||
run_eval=True,
|
run_eval=True,
|
||||||
|
@ -64,7 +63,6 @@ class TestFindUniquePhonemes(unittest.TestCase):
|
||||||
num_eval_loader_workers=0,
|
num_eval_loader_workers=0,
|
||||||
text_cleaner="english_cleaners",
|
text_cleaner="english_cleaners",
|
||||||
use_phonemes=True,
|
use_phonemes=True,
|
||||||
use_espeak_phonemes=False,
|
|
||||||
phoneme_language="en-us",
|
phoneme_language="en-us",
|
||||||
phoneme_cache_path="tests/data/ljspeech/phoneme_cache/",
|
phoneme_cache_path="tests/data/ljspeech/phoneme_cache/",
|
||||||
run_eval=True,
|
run_eval=True,
|
||||||
|
|
Loading…
Reference in New Issue