isort all imports

TTS/bin/compute_attention_masks.py

@@ -1,20 +1,20 @@
 import argparse
 import importlib
 import os
+from argparse import RawTextHelpFormatter
 
 import numpy as np
 import torch
 from torch.utils.data import DataLoader
 from tqdm import tqdm
-from argparse import RawTextHelpFormatter
+
 from TTS.tts.datasets.TTSDataset import MyDataset
 from TTS.tts.utils.generic_utils import setup_model
 from TTS.tts.utils.io import load_checkpoint
 from TTS.tts.utils.text.symbols import make_symbols, phonemes, symbols
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.io import load_config
 
-
 if __name__ == "__main__":
     # pylint: disable=bad-option-value
     parser = argparse.ArgumentParser(

TTS/bin/compute_embeddings.py

@@ -3,14 +3,14 @@
 import os
 
 import numpy as np
+import torch
 from tqdm import tqdm
 
-import torch
 from TTS.speaker_encoder.model import SpeakerEncoder
+from TTS.tts.datasets.preprocess import load_meta_data
+from TTS.tts.utils.speakers import save_speaker_mapping
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.io import load_config
-from TTS.tts.utils.speakers import save_speaker_mapping
-from TTS.tts.datasets.preprocess import load_meta_data
 
 parser = argparse.ArgumentParser(
     description='Compute embedding vectors for each wav file in a dataset. If "target_dataset" is defined, it generates "speakers.json" necessary for training a multi-speaker model.'

TTS/bin/compute_statistics.py

@@ -1,16 +1,16 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-import os
-import glob
 import argparse
+import glob
+import os
 
 import numpy as np
 from tqdm import tqdm
 
 from TTS.tts.datasets.preprocess import load_meta_data
-from TTS.utils.io import load_config
 from TTS.utils.audio import AudioProcessor
+from TTS.utils.io import load_config
 
 
 def main():

TTS/bin/convert_melgan_tflite.py

@@ -7,7 +7,6 @@
 from TTS.vocoder.tf.utils.io import load_checkpoint
 from TTS.vocoder.tf.utils.tflite import convert_melgan_to_tflite
 
-
 parser = argparse.ArgumentParser()
 parser.add_argument("--tf_model", type=str, help="Path to target torch model to be converted to TF.")
 parser.add_argument("--config_path", type=str, help="Path to config file of torch model.")

TTS/bin/convert_melgan_torch_to_tf.py

@@ -1,17 +1,14 @@
 import argparse
-from difflib import SequenceMatcher
 import os
+from difflib import SequenceMatcher
 
 import numpy as np
 import tensorflow as tf
 import torch
 
 from TTS.utils.io import load_config
-from TTS.vocoder.tf.utils.convert_torch_to_tf_utils import (
-    compare_torch_tf,
-    convert_tf_name,
-    transfer_weights_torch_to_tf,
-)
+from TTS.vocoder.tf.utils.convert_torch_to_tf_utils import (compare_torch_tf, convert_tf_name,
+                                                            transfer_weights_torch_to_tf)
 from TTS.vocoder.tf.utils.generic_utils import setup_generator as setup_tf_generator
 from TTS.vocoder.tf.utils.io import save_checkpoint
 from TTS.vocoder.utils.generic_utils import setup_generator

TTS/bin/convert_tacotron2_tflite.py

@@ -2,12 +2,11 @@
 
 import argparse
 
-from TTS.utils.io import load_config
-from TTS.tts.utils.text.symbols import symbols, phonemes
 from TTS.tts.tf.utils.generic_utils import setup_model
 from TTS.tts.tf.utils.io import load_checkpoint
 from TTS.tts.tf.utils.tflite import convert_tacotron2_to_tflite
-
+from TTS.tts.utils.text.symbols import phonemes, symbols
+from TTS.utils.io import load_config
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--tf_model", type=str, help="Path to target torch model to be converted to TF.")

TTS/bin/convert_tacotron2_torch_to_tf.py

@@ -1,12 +1,13 @@
 import argparse
-from difflib import SequenceMatcher
 import os
 import sys
+from difflib import SequenceMatcher
 from pprint import pprint
 
 import numpy as np
 import tensorflow as tf
 import torch
+
 from TTS.tts.tf.models.tacotron2 import Tacotron2
 from TTS.tts.tf.utils.convert_torch_to_tf_utils import compare_torch_tf, convert_tf_name, transfer_weights_torch_to_tf
 from TTS.tts.tf.utils.generic_utils import save_checkpoint

TTS/bin/distribute.py

@@ -1,12 +1,13 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
+import argparse
 import os
-import sys
 import pathlib
-import time
 import subprocess
-import argparse
+import sys
+import time
+
 import torch
 
 

TTS/bin/find_unique_chars.py

@@ -1,6 +1,6 @@
 """Find all the unique characters in a dataset"""
-import os
 import argparse
+import os
 from argparse import RawTextHelpFormatter
 
 from TTS.tts.datasets.preprocess import get_preprocessor_by_name

TTS/bin/resample.py

@@ -1,10 +1,11 @@
 import argparse
 import glob
 import os
-import librosa
-from distutils.dir_util import copy_tree
 from argparse import RawTextHelpFormatter
+from distutils.dir_util import copy_tree
 from multiprocessing import Pool
+
+import librosa
 from tqdm import tqdm
 
 

TTS/bin/synthesize.py

@@ -4,7 +4,6 @@
 import argparse
 import sys
 from argparse import RawTextHelpFormatter
-
 # pylint: disable=redefined-outer-name, unused-argument
 from pathlib import Path
 

TTS/bin/train_align_tts.py

@@ -12,6 +12,7 @@
 from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
+
 from TTS.tts.datasets.preprocess import load_meta_data
 from TTS.tts.datasets.TTSDataset import MyDataset
 from TTS.tts.layers.losses import AlignTTSLoss

TTS/bin/train_encoder.py

@@ -9,20 +9,16 @@
 
 import torch
 from torch.utils.data import DataLoader
+
 from TTS.speaker_encoder.dataset import MyDataset
 from TTS.speaker_encoder.losses import AngleProtoLoss, GE2ELoss
 from TTS.speaker_encoder.model import SpeakerEncoder
 from TTS.speaker_encoder.utils.generic_utils import check_config_speaker_encoder, save_best_model
 from TTS.speaker_encoder.utils.visual import plot_embeddings
 from TTS.tts.datasets.preprocess import load_meta_data
 from TTS.utils.audio import AudioProcessor
-from TTS.utils.generic_utils import (
-    count_parameters,
-    create_experiment_folder,
-    get_git_branch,
-    remove_experiment_folder,
-    set_init_dict,
-)
+from TTS.utils.generic_utils import (count_parameters, create_experiment_folder, get_git_branch,
+                                     remove_experiment_folder, set_init_dict)
 from TTS.utils.io import copy_model_files, load_config
 from TTS.utils.radam import RAdam
 from TTS.utils.tensorboard_logger import TensorboardLogger

TTS/bin/train_glow_tts.py

@@ -8,13 +8,11 @@
 from random import randrange
 
 import torch
-
 # DISTRIBUTED
 from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
 
-from TTS.utils.arguments import parse_arguments, process_args
 from TTS.tts.datasets.preprocess import load_meta_data
 from TTS.tts.datasets.TTSDataset import MyDataset
 from TTS.tts.layers.losses import GlowTTSLoss
@@ -25,6 +23,7 @@
 from TTS.tts.utils.synthesis import synthesis
 from TTS.tts.utils.text.symbols import make_symbols, phonemes, symbols
 from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
+from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.distribute import init_distributed, reduce_tensor
 from TTS.utils.generic_utils import KeepAverage, count_parameters, remove_experiment_folder, set_init_dict

TTS/bin/train_speedy_speech.py

@@ -5,16 +5,15 @@
 import sys
 import time
 import traceback
-import numpy as np
 from random import randrange
 
+import numpy as np
 import torch
-from TTS.utils.arguments import parse_arguments, process_args
-
 # DISTRIBUTED
 from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
+
 from TTS.tts.datasets.preprocess import load_meta_data
 from TTS.tts.datasets.TTSDataset import MyDataset
 from TTS.tts.layers.losses import SpeedySpeechLoss
@@ -25,6 +24,7 @@
 from TTS.tts.utils.synthesis import synthesis
 from TTS.tts.utils.text.symbols import make_symbols, phonemes, symbols
 from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
+from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.distribute import init_distributed, reduce_tensor
 from TTS.utils.generic_utils import KeepAverage, count_parameters, remove_experiment_folder, set_init_dict

TTS/bin/train_tacotron.py

@@ -10,7 +10,7 @@
 import numpy as np
 import torch
 from torch.utils.data import DataLoader
-from TTS.utils.arguments import parse_arguments, process_args
+
 from TTS.tts.datasets.preprocess import load_meta_data
 from TTS.tts.datasets.TTSDataset import MyDataset
 from TTS.tts.layers.losses import TacotronLoss
@@ -21,18 +21,13 @@
 from TTS.tts.utils.synthesis import synthesis
 from TTS.tts.utils.text.symbols import make_symbols, phonemes, symbols
 from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
+from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.distribute import DistributedSampler, apply_gradient_allreduce, init_distributed, reduce_tensor
 from TTS.utils.generic_utils import KeepAverage, count_parameters, remove_experiment_folder, set_init_dict
 from TTS.utils.radam import RAdam
-from TTS.utils.training import (
-    NoamLR,
-    adam_weight_decay,
-    check_update,
-    gradual_training_scheduler,
-    set_weight_decay,
-    setup_torch_training_env,
-)
+from TTS.utils.training import (NoamLR, adam_weight_decay, check_update, gradual_training_scheduler, set_weight_decay,
+                                setup_torch_training_env)
 
 use_cuda, num_gpus = setup_torch_training_env(True, False)
 

TTS/bin/train_vocoder_gan.py

@@ -9,25 +9,23 @@
 from inspect import signature
 
 import torch
+# DISTRIBUTED
+from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.utils.data import DataLoader
+from torch.utils.data.distributed import DistributedSampler
+
 from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
+from TTS.utils.distribute import init_distributed
 from TTS.utils.generic_utils import KeepAverage, count_parameters, remove_experiment_folder, set_init_dict
-
 from TTS.utils.radam import RAdam
-
 from TTS.utils.training import setup_torch_training_env
 from TTS.vocoder.datasets.gan_dataset import GANDataset
 from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data
 from TTS.vocoder.layers.losses import DiscriminatorLoss, GeneratorLoss
 from TTS.vocoder.utils.generic_utils import plot_results, setup_discriminator, setup_generator
 from TTS.vocoder.utils.io import save_best_model, save_checkpoint
 
-# DISTRIBUTED
-from torch.nn.parallel import DistributedDataParallel as DDP_th
-from torch.utils.data.distributed import DistributedSampler
-from TTS.utils.distribute import init_distributed
-
 use_cuda, num_gpus = setup_torch_training_env(True, True)
 
 

TTS/bin/train_vocoder_wavegrad.py

@@ -5,15 +5,15 @@
 import sys
 import time
 import traceback
-import numpy as np
 
+import numpy as np
 import torch
-
 # DISTRIBUTED
 from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.optim import Adam
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
+
 from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.distribute import init_distributed

TTS/bin/train_vocoder_wavernn.py

@@ -2,33 +2,29 @@
 """Train WaveRNN vocoder model."""
 
 import os
+import random
 import sys
-import traceback
 import time
-import random
+import traceback
 
 import torch
 from torch.utils.data import DataLoader
 
-# from torch.utils.data.distributed import DistributedSampler
-
-from TTS.utils.arguments import parse_arguments, process_args
 from TTS.tts.utils.visual import plot_spectrogram
+from TTS.utils.arguments import parse_arguments, process_args
 from TTS.utils.audio import AudioProcessor
+from TTS.utils.generic_utils import KeepAverage, count_parameters, remove_experiment_folder, set_init_dict
 from TTS.utils.radam import RAdam
 from TTS.utils.training import setup_torch_training_env
-from TTS.utils.generic_utils import (
-    KeepAverage,
-    count_parameters,
-    remove_experiment_folder,
-    set_init_dict,
-)
-from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset
 from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data
+from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset
 from TTS.vocoder.utils.distribution import discretized_mix_logistic_loss, gaussian_loss
 from TTS.vocoder.utils.generic_utils import setup_generator
 from TTS.vocoder.utils.io import save_best_model, save_checkpoint
 
+# from torch.utils.data.distributed import DistributedSampler
+
+
 
 use_cuda, num_gpus = setup_torch_training_env(True, True)
 

TTS/bin/tune_wavegrad.py

@@ -6,6 +6,7 @@
 import torch
 from torch.utils.data import DataLoader
 from tqdm import tqdm
+
 from TTS.utils.audio import AudioProcessor
 from TTS.utils.io import load_config
 from TTS.vocoder.datasets.preprocess import load_wav_data

TTS/server/server.py

@@ -1,14 +1,15 @@
 #!flask/bin/python
 import argparse
+import io
 import os
 import sys
-import io
 from pathlib import Path
 
 from flask import Flask, render_template, request, send_file
-from TTS.utils.synthesizer import Synthesizer
-from TTS.utils.manage import ModelManager
+
 from TTS.utils.io import load_config
+from TTS.utils.manage import ModelManager
+from TTS.utils.synthesizer import Synthesizer
 
 
 def create_argparser():