mirror of https://github.com/coqui-ai/TTS.git
correct import statements for tests
This commit is contained in:
parent
6488d5e305
commit
0b35e8e949
|
@ -2,8 +2,8 @@ import os
|
||||||
import unittest
|
import unittest
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch as T
|
import torch as T
|
||||||
from TTS.utils.audio import AudioProcessor
|
from utils.audio import AudioProcessor
|
||||||
from TTS.utils.generic_utils import load_config
|
from utils.generic_utils import load_config
|
||||||
|
|
||||||
file_path = os.path.dirname(os.path.realpath(__file__))
|
file_path = os.path.dirname(os.path.realpath(__file__))
|
||||||
INPUTPATH = os.path.join(file_path, 'inputs')
|
INPUTPATH = os.path.join(file_path, 'inputs')
|
||||||
|
|
|
@ -1,8 +1,8 @@
|
||||||
import unittest
|
import unittest
|
||||||
import torch as T
|
import torch as T
|
||||||
|
|
||||||
from TTS.utils.generic_utils import save_checkpoint, save_best_model
|
from utils.generic_utils import save_checkpoint, save_best_model
|
||||||
from TTS.layers.tacotron import Prenet, CBHG, Decoder, Encoder
|
from layers.tacotron import Prenet, CBHG, Decoder, Encoder
|
||||||
|
|
||||||
OUT_PATH = '/tmp/test.pth.tar'
|
OUT_PATH = '/tmp/test.pth.tar'
|
||||||
|
|
||||||
|
|
|
@ -1,9 +1,9 @@
|
||||||
import unittest
|
import unittest
|
||||||
import torch as T
|
import torch as T
|
||||||
|
|
||||||
from TTS.layers.tacotron import Prenet, CBHG, Decoder, Encoder
|
from layers.tacotron import Prenet, CBHG, Decoder, Encoder
|
||||||
from TTS.layers.losses import L1LossMasked
|
from layers.losses import L1LossMasked
|
||||||
from TTS.utils.generic_utils import sequence_mask
|
from utils.generic_utils import sequence_mask
|
||||||
|
|
||||||
|
|
||||||
class PrenetTests(unittest.TestCase):
|
class PrenetTests(unittest.TestCase):
|
||||||
|
|
|
@ -4,10 +4,10 @@ import shutil
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
from torch.utils.data import DataLoader
|
from torch.utils.data import DataLoader
|
||||||
from TTS.utils.generic_utils import load_config
|
from utils.generic_utils import load_config
|
||||||
from TTS.utils.audio import AudioProcessor
|
from utils.audio import AudioProcessor
|
||||||
from TTS.datasets import TTSDataset, TTSDatasetCached, TTSDatasetMemory
|
from datasets import TTSDataset, TTSDatasetCached, TTSDatasetMemory
|
||||||
from TTS.datasets.preprocess import ljspeech, tts_cache
|
from datasets.preprocess import ljspeech, tts_cache
|
||||||
|
|
||||||
file_path = os.path.dirname(os.path.realpath(__file__))
|
file_path = os.path.dirname(os.path.realpath(__file__))
|
||||||
OUTPATH = os.path.join(file_path, "outputs/loader_tests/")
|
OUTPATH = os.path.join(file_path, "outputs/loader_tests/")
|
||||||
|
|
|
@ -6,9 +6,9 @@ import numpy as np
|
||||||
|
|
||||||
from torch import optim
|
from torch import optim
|
||||||
from torch import nn
|
from torch import nn
|
||||||
from TTS.utils.generic_utils import load_config
|
from utils.generic_utils import load_config
|
||||||
from TTS.layers.losses import L1LossMasked
|
from layers.losses import L1LossMasked
|
||||||
from TTS.models.tacotron import Tacotron
|
from models.tacotron import Tacotron
|
||||||
|
|
||||||
torch.manual_seed(1)
|
torch.manual_seed(1)
|
||||||
use_cuda = torch.cuda.is_available()
|
use_cuda = torch.cuda.is_available()
|
||||||
|
|
Loading…
Reference in New Issue