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 import numpy as np
from torch import nn from torch import nn
from TTS.config import load_config
from TTS.utils.audio.numpy_transforms import save_wav from TTS.utils.audio.numpy_transforms import save_wav
from TTS.utils.manage import ModelManager from TTS.utils.manage import ModelManager
from TTS.utils.synthesizer import Synthesizer from TTS.utils.synthesizer import Synthesizer
from TTS.config import load_config
class TTS(nn.Module): class TTS(nn.Module):

View File

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

View File

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

View File

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

View File

@ -1,19 +1,16 @@
import argparse import argparse
import logging
import os import os
import sys import sys
import tempfile import tempfile
import traceback
import gradio as gr import gradio as gr
import librosa.display
import numpy as np
import os
import torch import torch
import torchaudio import torchaudio
import traceback
from TTS.demos.xtts_ft_demo.utils.formatter import format_audio_list 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.demos.xtts_ft_demo.utils.gpt_train import train_gpt
from TTS.tts.configs.xtts_config import XttsConfig from TTS.tts.configs.xtts_config import XttsConfig
from TTS.tts.models.xtts import Xtts from TTS.tts.models.xtts import Xtts
@ -91,7 +88,7 @@ sys.stderr = sys.stdout
# logging.basicConfig(stream=sys.stdout, level=logging.INFO) # logging.basicConfig(stream=sys.stdout, level=logging.INFO)
import logging
logging.basicConfig( logging.basicConfig(
level=logging.INFO, level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(message)s", 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 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 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 # Modified code from https://github.com/lucidrains/audiolm-pytorch/blob/main/audiolm_pytorch/hubert_kmeans.py
import logging
from pathlib import Path
import torch import torch
from einops import pack, unpack from einops import pack, unpack

View File

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

View File

@ -2,7 +2,7 @@ import torch
from torch import nn from torch import nn
from torch.nn.modules.conv import Conv1d 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): class DiscriminatorS(torch.nn.Module):

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@ -14,7 +14,7 @@ from TTS.model import BaseTrainerModel
from TTS.tts.datasets.dataset import TTSDataset from TTS.tts.datasets.dataset import TTSDataset
from TTS.tts.utils.data import get_length_balancer_weights 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.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.synthesis import synthesis
from TTS.tts.utils.visual import plot_alignment, plot_spectrogram from TTS.tts.utils.visual import plot_alignment, plot_spectrogram

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@ -6,10 +6,10 @@ import numpy as np
import torch import torch
from torch.utils.data import DataLoader 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 tests import get_tests_data_path, get_tests_output_path
from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig
from TTS.tts.datasets import load_tts_samples 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.tts.utils.text.tokenizer import TTSTokenizer
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor

View File

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