diff --git a/tests/audio_tests.py b/tests/audio_tests.py index fb5f698e..8c432d2f 100644 --- a/tests/audio_tests.py +++ b/tests/audio_tests.py @@ -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') diff --git a/tests/generic_utils_text.py b/tests/generic_utils_text.py index 63b8bc1a..56b794ce 100644 --- a/tests/generic_utils_text.py +++ b/tests/generic_utils_text.py @@ -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' diff --git a/tests/layers_tests.py b/tests/layers_tests.py index 2458e00b..be6a8516 100644 --- a/tests/layers_tests.py +++ b/tests/layers_tests.py @@ -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): diff --git a/tests/loader_tests.py b/tests/loader_tests.py index e5dae929..c945a592 100644 --- a/tests/loader_tests.py +++ b/tests/loader_tests.py @@ -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/") diff --git a/tests/tacotron_tests.py b/tests/tacotron_tests.py index 0e872a07..bcb702b2 100644 --- a/tests/tacotron_tests.py +++ b/tests/tacotron_tests.py @@ -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()