Fix imports

This commit is contained in:
Eren Gölge 2021-09-30 14:28:28 +00:00
parent 16b70be0dd
commit 9f23ad6a0f
4 changed files with 8 additions and 8 deletions

View File

@ -6,7 +6,6 @@ import torch
from coqpit import Coqpit
from torch import nn
from TTS.utils.audio import AudioProcessor
# pylint: skip-file

View File

@ -308,14 +308,14 @@ def mls(root_path, meta_files=None):
# ======================================== VOX CELEB ===========================================
def voxceleb2(root_path, meta_file=None):
"""
:param meta_file Used only for consistency with load_meta_data api
:param meta_file Used only for consistency with load_tts_samples api
"""
return _voxcel_x(root_path, meta_file, voxcel_idx="2")
def voxceleb1(root_path, meta_file=None):
"""
:param meta_file Used only for consistency with load_meta_data api
:param meta_file Used only for consistency with load_tts_samples api
"""
return _voxcel_x(root_path, meta_file, voxcel_idx="1")

View File

@ -110,10 +110,10 @@ class SpeakerManager:
@staticmethod
def parse_speakers_from_data(items: list) -> Tuple[Dict, int]:
"""Parse speaker IDs from data samples retured by `load_meta_data()`.
"""Parse speaker IDs from data samples retured by `load_tts_samples()`.
Args:
items (list): Data sampled returned by `load_meta_data()`.
items (list): Data sampled returned by `load_tts_samples()`.
Returns:
Tuple[Dict, int]: speaker IDs and number of speakers.
@ -127,7 +127,7 @@ class SpeakerManager:
"""Set speaker IDs from data samples.
Args:
items (List): Data sampled returned by `load_meta_data()`.
items (List): Data sampled returned by `load_tts_samples()`.
"""
self.speaker_ids, _ = self.parse_speakers_from_data(items)

View File

@ -5,6 +5,7 @@ from torch.utils.data import Dataset
from TTS.utils.audio import AudioProcessor
from TTS.vocoder.datasets.gan_dataset import GANDataset
from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data
from TTS.vocoder.datasets.wavegrad_dataset import WaveGradDataset
from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset