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 coqpit import Coqpit
from torch import nn from torch import nn
from TTS.utils.audio import AudioProcessor
# pylint: skip-file # pylint: skip-file

View File

@ -308,14 +308,14 @@ def mls(root_path, meta_files=None):
# ======================================== VOX CELEB =========================================== # ======================================== VOX CELEB ===========================================
def voxceleb2(root_path, meta_file=None): 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") return _voxcel_x(root_path, meta_file, voxcel_idx="2")
def voxceleb1(root_path, meta_file=None): 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") return _voxcel_x(root_path, meta_file, voxcel_idx="1")

View File

@ -110,10 +110,10 @@ class SpeakerManager:
@staticmethod @staticmethod
def parse_speakers_from_data(items: list) -> Tuple[Dict, int]: 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: Args:
items (list): Data sampled returned by `load_meta_data()`. items (list): Data sampled returned by `load_tts_samples()`.
Returns: Returns:
Tuple[Dict, int]: speaker IDs and number of speakers. Tuple[Dict, int]: speaker IDs and number of speakers.
@ -127,7 +127,7 @@ class SpeakerManager:
"""Set speaker IDs from data samples. """Set speaker IDs from data samples.
Args: 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) 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.utils.audio import AudioProcessor
from TTS.vocoder.datasets.gan_dataset import GANDataset 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.wavegrad_dataset import WaveGradDataset
from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset