Update tests

pull/1324/head
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
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
def test_make_symbols():
_ = create_phonemes()
_ = create_graphemes()
class BaseCharacterTest(unittest.TestCase):
def setUp(self):
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):
@dataclass
class Characters(Coqpit):
characters_class: str = None
characters: str = _phonemes
punctuations: str = _punctuations
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.utils.speakers import SpeakerManager
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

View File

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