coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

correct import statements for tests

pull/10/head
Eren Golge 2018-12-11 15:07:50 +01:00
parent 6488d5e305
commit 0b35e8e949
5 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/audio_tests.py
View File

@ -2,8 +2,8 @@ import os
import unittest
import numpy as np
import torch as T
from TTS.utils.audio import AudioProcessor
from TTS.utils.generic_utils import load_config
from utils.audio import AudioProcessor
from utils.generic_utils import load_config
file_path = os.path.dirname(os.path.realpath(__file__))
INPUTPATH = os.path.join(file_path, 'inputs')

4
tests/generic_utils_text.py
View File

@ -1,8 +1,8 @@
import unittest
import torch as T
from TTS.utils.generic_utils import save_checkpoint, save_best_model
from TTS.layers.tacotron import Prenet, CBHG, Decoder, Encoder
from utils.generic_utils import save_checkpoint, save_best_model
from layers.tacotron import Prenet, CBHG, Decoder, Encoder
OUT_PATH = '/tmp/test.pth.tar'

6
tests/layers_tests.py
View File

@ -1,9 +1,9 @@
import unittest
import torch as T
from TTS.layers.tacotron import Prenet, CBHG, Decoder, Encoder
from TTS.layers.losses import L1LossMasked
from TTS.utils.generic_utils import sequence_mask
from layers.tacotron import Prenet, CBHG, Decoder, Encoder
from layers.losses import L1LossMasked
from utils.generic_utils import sequence_mask
class PrenetTests(unittest.TestCase):

8
tests/loader_tests.py
View File

@ -4,10 +4,10 @@ import shutil
import numpy as np
from torch.utils.data import DataLoader
from TTS.utils.generic_utils import load_config
from TTS.utils.audio import AudioProcessor
from TTS.datasets import TTSDataset, TTSDatasetCached, TTSDatasetMemory
from TTS.datasets.preprocess import ljspeech, tts_cache
from utils.generic_utils import load_config
from utils.audio import AudioProcessor
from datasets import TTSDataset, TTSDatasetCached, TTSDatasetMemory
from datasets.preprocess import ljspeech, tts_cache
file_path = os.path.dirname(os.path.realpath(__file__))
OUTPATH = os.path.join(file_path, "outputs/loader_tests/")

6
tests/tacotron_tests.py
View File

@ -6,9 +6,9 @@ import numpy as np
from torch import optim
from torch import nn
from TTS.utils.generic_utils import load_config
from TTS.layers.losses import L1LossMasked
from TTS.models.tacotron import Tacotron
from utils.generic_utils import load_config
from layers.losses import L1LossMasked
from models.tacotron import Tacotron
torch.manual_seed(1)
use_cuda = torch.cuda.is_available()