Update tests

This commit is contained in:
Eren Gölge 2022-02-20 11:31:45 +01:00
parent 1e219fef0a
commit b0cff949f5
4 changed files with 4 additions and 7 deletions

View File

@ -1,15 +1,10 @@
import unittest import unittest
from TTS.tts.utils.text.characters import BaseCharacters, Graphemes, IPAPhonemes, create_graphemes, create_phonemes from TTS.tts.utils.text.characters import BaseCharacters, Graphemes, IPAPhonemes
# pylint: disable=protected-access # pylint: disable=protected-access
def test_make_symbols():
_ = create_phonemes()
_ = create_graphemes()
class BaseCharacterTest(unittest.TestCase): class BaseCharacterTest(unittest.TestCase):
def setUp(self): def setUp(self):
self.characters_empty = BaseCharacters("", "", pad="", eos="", bos="", blank="", is_unique=True, is_sorted=True) self.characters_empty = BaseCharacters("", "", pad="", eos="", bos="", blank="", is_unique=True, is_sorted=True)

View File

@ -64,6 +64,7 @@ class TestTTSTokenizer(unittest.TestCase):
def test_init_from_config(self): def test_init_from_config(self):
@dataclass @dataclass
class Characters(Coqpit): class Characters(Coqpit):
characters_class: str = None
characters: str = _phonemes characters: str = _phonemes
punctuations: str = _punctuations punctuations: str = _punctuations
pad: str = _pad pad: str = _pad

View File

@ -11,7 +11,7 @@ from TTS.tts.layers.losses import GlowTTSLoss
from TTS.tts.models.glow_tts import GlowTTS from TTS.tts.models.glow_tts import GlowTTS
from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.speakers import SpeakerManager
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.logging.tensorboard_logger import TensorboardLogger from trainer.logging.tensorboard_logger import TensorboardLogger
# pylint: disable=unused-variable # pylint: disable=unused-variable

View File

@ -20,6 +20,7 @@ config = MultibandMelganConfig(
eval_split_size=1, eval_split_size=1,
print_step=1, print_step=1,
print_eval=True, print_eval=True,
steps_to_start_discriminator=1,
data_path="tests/data/ljspeech", data_path="tests/data/ljspeech",
discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]}, discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]},
output_path=output_path, output_path=output_path,