diff --git a/tests/data_tests/test_loader.py b/tests/data_tests/test_loader.py index 0562fbf7..6790fd54 100644 --- a/tests/data_tests/test_loader.py +++ b/tests/data_tests/test_loader.py @@ -6,7 +6,7 @@ import numpy as np import torch from torch.utils.data import DataLoader -from tests import get_tests_output_path +from tests import get_tests_data_path, get_tests_output_path from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig from TTS.tts.datasets import TTSDataset, load_tts_samples from TTS.tts.utils.text.tokenizer import TTSTokenizer @@ -20,7 +20,7 @@ os.makedirs(OUTPATH, exist_ok=True) # create a dummy config for testing data loaders. c = BaseTTSConfig(text_cleaner="english_cleaners", num_loader_workers=0, batch_size=2, use_noise_augment=False) c.r = 5 -c.data_path = "tests/data/ljspeech/" +c.data_path = os.path.join(get_tests_data_path(), "ljspeech/") ok_ljspeech = os.path.exists(c.data_path) dataset_config = BaseDatasetConfig(