Ruff autofix unused imports and import order

This commit is contained in:
Aarni Koskela 2023-09-27 00:42:14 +03:00
parent 4f859824ac
commit 90991e89b4
25 changed files with 23 additions and 54 deletions

View File

@ -6,10 +6,10 @@ from typing import Union
import numpy as np
from torch import nn
from TTS.config import load_config
from TTS.utils.audio.numpy_transforms import save_wav
from TTS.utils.manage import ModelManager
from TTS.utils.synthesizer import Synthesizer
from TTS.config import load_config
class TTS(nn.Module):

View File

@ -1,4 +1,5 @@
"""Get detailed info about the working environment."""
import json
import os
import platform
import sys
@ -6,11 +7,10 @@ import sys
import numpy
import torch
sys.path += [os.path.abspath(".."), os.path.abspath(".")]
import json
import TTS
sys.path += [os.path.abspath(".."), os.path.abspath(".")]
def system_info():
return {

View File

@ -1,23 +1,17 @@
import os
import gc
import torchaudio
import os
import pandas
from faster_whisper import WhisperModel
from glob import glob
from tqdm import tqdm
import torch
import torchaudio
# torch.set_num_threads(1)
from faster_whisper import WhisperModel
from tqdm import tqdm
# torch.set_num_threads(1)
from TTS.tts.layers.xtts.tokenizer import multilingual_cleaners
torch.set_num_threads(16)
import os
audio_types = (".wav", ".mp3", ".flac")

View File

@ -1,5 +1,5 @@
import os
import gc
import os
from trainer import Trainer, TrainerArgs

View File

@ -1,19 +1,16 @@
import argparse
import logging
import os
import sys
import tempfile
import traceback
import gradio as gr
import librosa.display
import numpy as np
import os
import torch
import torchaudio
import traceback
from TTS.demos.xtts_ft_demo.utils.formatter import format_audio_list
from TTS.demos.xtts_ft_demo.utils.gpt_train import train_gpt
from TTS.tts.configs.xtts_config import XttsConfig
from TTS.tts.models.xtts import Xtts
@ -91,7 +88,7 @@ sys.stderr = sys.stdout
# logging.basicConfig(stream=sys.stdout, level=logging.INFO)
import logging
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(message)s",

View File

@ -1,4 +1,4 @@
from dataclasses import asdict, dataclass
from dataclasses import dataclass
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig

View File

@ -1,4 +1,4 @@
from dataclasses import asdict, dataclass
from dataclasses import dataclass
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig

View File

@ -7,8 +7,6 @@ License: MIT
# Modified code from https://github.com/lucidrains/audiolm-pytorch/blob/main/audiolm_pytorch/hubert_kmeans.py
import logging
from pathlib import Path
import torch
from einops import pack, unpack

View File

@ -1,6 +1,5 @@
import functools
import math
import os
import fsspec
import torch

View File

@ -2,7 +2,7 @@ import torch
from torch import nn
from torch.nn.modules.conv import Conv1d
from TTS.vocoder.models.hifigan_discriminator import DiscriminatorP, MultiPeriodDiscriminator
from TTS.vocoder.models.hifigan_discriminator import DiscriminatorP
class DiscriminatorS(torch.nn.Module):

View File

@ -1,7 +1,6 @@
# ported from: https://github.com/neonbjb/tortoise-tts
import functools
import math
import random
import torch

View File

@ -1,5 +1,3 @@
import math
import torch
from torch import nn
from transformers import GPT2PreTrainedModel

View File

@ -1,4 +1,3 @@
import os
import random
import sys

View File

@ -5,7 +5,6 @@ import torch
import torch.nn as nn
import torchaudio
from coqpit import Coqpit
from torch.nn import functional as F
from torch.utils.data import DataLoader
from trainer.torch import DistributedSampler
from trainer.trainer_utils import get_optimizer, get_scheduler

View File

@ -4,13 +4,11 @@
import argparse
import csv
import os
import re
import string
import sys
# fmt: off
# ================================================================================ #
# basic constant
# ================================================================================ #

View File

@ -14,7 +14,7 @@ from TTS.model import BaseTrainerModel
from TTS.tts.datasets.dataset import TTSDataset
from TTS.tts.utils.data import get_length_balancer_weights
from TTS.tts.utils.languages import LanguageManager, get_language_balancer_weights
from TTS.tts.utils.speakers import SpeakerManager, get_speaker_balancer_weights, get_speaker_manager
from TTS.tts.utils.speakers import SpeakerManager, get_speaker_balancer_weights
from TTS.tts.utils.synthesis import synthesis
from TTS.tts.utils.visual import plot_alignment, plot_spectrogram

View File

@ -1,7 +1,5 @@
from dataclasses import asdict, dataclass, field
from typing import Dict, List
from coqpit import Coqpit, check_argument
from dataclasses import dataclass, field
from typing import List
from TTS.config import BaseAudioConfig, BaseDatasetConfig, BaseTrainingConfig

View File

@ -1,8 +1,6 @@
import math
import numpy as np
import torch
from torch import nn
from torch.nn import functional as F

View File

@ -1,11 +1,9 @@
import struct
from pathlib import Path
from typing import Optional, Union
# import webrtcvad
import librosa
import numpy as np
from scipy.ndimage.morphology import binary_dilation
from TTS.vc.modules.freevc.speaker_encoder.hparams import (
audio_norm_target_dBFS,

View File

@ -1,4 +1,3 @@
from pathlib import Path
from time import perf_counter as timer
from typing import List, Union

View File

@ -1,6 +1,3 @@
import os
from coqpit import Coqpit
from trainer import Trainer, TrainerArgs
from TTS.tts.configs.shared_configs import BaseAudioConfig

View File

@ -4,7 +4,6 @@ import torch
from trainer import Trainer, TrainerArgs
from TTS.bin.compute_embeddings import compute_embeddings
from TTS.bin.resample import resample_files
from TTS.config.shared_configs import BaseDatasetConfig
from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.datasets import load_tts_samples

View File

@ -23,12 +23,12 @@
import os
import subprocess
import sys
from packaging.version import Version
import numpy
import setuptools.command.build_py
import setuptools.command.develop
from Cython.Build import cythonize
from packaging.version import Version
from setuptools import Extension, find_packages, setup
python_version = sys.version.split()[0]

View File

@ -6,10 +6,10 @@ import numpy as np
import torch
from torch.utils.data import DataLoader
from TTS.tts.datasets.dataset import TTSDataset
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 load_tts_samples
from TTS.tts.datasets.dataset import TTSDataset
from TTS.tts.utils.text.tokenizer import TTSTokenizer
from TTS.utils.audio import AudioProcessor

View File

@ -4,8 +4,7 @@ import unittest
import torch
from tests import get_tests_input_path
from TTS.vc.configs.freevc_config import FreeVCConfig
from TTS.vc.models.freevc import FreeVC
from TTS.vc.models.freevc import FreeVC, FreeVCConfig
# pylint: disable=unused-variable
# pylint: disable=no-self-use