Unverified Commit 0f23e6d3 authored by Krishna Kalyan's avatar Krishna Kalyan Committed by GitHub
Browse files

Improve Flake Rules (#1214)

parent cd5a1247
[flake8] [flake8]
max-line-length = 120 max-line-length = 120
ignore = E305,E402,E721,E741,F401,F403,F405,W503,W504,F999 ignore = E305,E402,E721,E741,F405,W503,W504,F999
exclude = build,docs/source,_ext,third_party exclude = build,docs/source,_ext,third_party
import io import io
import itertools
from unittest.mock import patch from unittest.mock import patch
from torchaudio._internal import module_utils as _mod_utils from torchaudio._internal import module_utils as _mod_utils
......
"""Test numerical consistency among single input and batched input.""" """Test numerical consistency among single input and batched input."""
import unittest
import itertools import itertools
from parameterized import parameterized from parameterized import parameterized
......
...@@ -27,4 +27,8 @@ from .wav_utils import ( ...@@ -27,4 +27,8 @@ from .wav_utils import (
from .parameterized_utils import ( from .parameterized_utils import (
load_params, load_params,
) )
from . import sox_utils
__all__ = ['get_asset_path', 'get_whitenoise', 'get_sinusoid', 'set_audio_backend',
'TempDirMixin', 'HttpServerMixin', 'TestBaseMixin', 'PytorchTestCase', 'TorchaudioTestCase',
'skipIfNoCuda', 'skipIfNoExec', 'skipIfNoModule', 'skipIfNoExtension', 'skipIfNoSoxBackend',
'get_wav_data', 'normalize_wav', 'load_wav', 'save_wav', 'load_params']
import os import os
import math import math
import unittest
import torch import torch
import torchaudio import torchaudio
......
from torchaudio.datasets.librispeech import LIBRISPEECH
from torchaudio.datasets.speechcommands import SPEECHCOMMANDS
from torchaudio.datasets.vctk import VCTK from torchaudio.datasets.vctk import VCTK
from torchaudio_unittest.common_utils import ( from torchaudio_unittest.common_utils import (
......
import unittest
import torch import torch
import torchaudio.kaldi_io as kio import torchaudio.kaldi_io as kio
......
import unittest
import torch import torch
import torchaudio.functional as F import torchaudio.functional as F
import torchaudio.transforms as T import torchaudio.transforms as T
......
...@@ -10,7 +10,6 @@ from torchaudio_unittest.common_utils import ( ...@@ -10,7 +10,6 @@ from torchaudio_unittest.common_utils import (
save_wav, save_wav,
) )
from .common import ( from .common import (
name_func,
load_params, load_params,
) )
......
"""Test suites for jit-ability and its numerical compatibility""" """Test suites for jit-ability and its numerical compatibility"""
import unittest
import torch import torch
import torchaudio.transforms as T import torchaudio.transforms as T
......
import math import math
import unittest
import torch import torch
import torchaudio import torchaudio
......
...@@ -30,3 +30,9 @@ try: ...@@ -30,3 +30,9 @@ try:
from .version import __version__, git_version # noqa: F401 from .version import __version__, git_version # noqa: F401
except ImportError: except ImportError:
pass pass
__all__ = ['extension', '_mod_utils', 'compliance', 'datasets', 'functional', 'kaldi_io',
'utils', 'sox_effects', 'transforms', 'list_audio_backends',
'get_audio_backend', 'set_audio_backend', 'save_encinfo', 'sox_signalinfo_t',
'sox_encodinginfo_t', 'get_sox_option_t', 'get_sox_encoding_t',
'get_sox_bool', 'SignalInfo', 'EncodingInfo']
# flake8: noqa
from . import utils from . import utils
from .utils import ( from .utils import (
list_audio_backends, list_audio_backends,
......
from . import kaldi from . import kaldi
__all__ = ['kaldi']
import os import os
import warnings
from pathlib import Path from pathlib import Path
from typing import Any, Tuple, Optional, Union from typing import Tuple, Optional, Union
import torchaudio import torchaudio
from torch import Tensor from torch import Tensor
......
import os import os
import csv import csv
from typing import List, Tuple, Union from typing import Tuple, Union
from pathlib import Path from pathlib import Path
import torchaudio import torchaudio
......
import warnings import warnings
import importlib import importlib
from collections import namedtuple
import torch import torch
from torchaudio._internal import module_utils as _mod_utils from torchaudio._internal import module_utils as _mod_utils
......
...@@ -41,3 +41,11 @@ from .filtering import ( ...@@ -41,3 +41,11 @@ from .filtering import (
treble_biquad, treble_biquad,
vad, vad,
) )
__all__ = ['amplitude_to_DB', 'angle', 'complex_norm', 'compute_deltas', 'create_dct',
'create_fb_matrix', 'DB_to_amplitude', 'detect_pitch_frequency', 'griffinlim', 'magphase',
'mask_along_axis', 'mask_along_axis_iid', 'mu_law_encoding', 'mu_law_decoding', 'phase_vocoder',
'sliding_window_cmn', 'spectrogram', 'spectral_centroid', 'allpass_biquad', 'band_biquad',
'bandpass_biquad', 'bandreject_biquad', 'bass_biquad', 'biquad', 'contrast', 'dither',
'dcshift', 'deemph_biquad', 'equalizer_biquad', 'flanger', 'gain', 'highpass_biquad',
'lfilter', 'lowpass_biquad', 'overdrive', 'phaser', 'riaa_biquad', 'treble_biquad', 'vad']
from .wav2letter import Wav2Letter from .wav2letter import Wav2Letter
from .wavernn import WaveRNN from .wavernn import WaveRNN
from .conv_tasnet import ConvTasNet from .conv_tasnet import ConvTasNet
__all__ = ['Wav2Letter', 'WaveRNN', 'ConvTasNet']
...@@ -12,3 +12,5 @@ if _mod_utils.is_module_available('torchaudio._torchaudio'): ...@@ -12,3 +12,5 @@ if _mod_utils.is_module_available('torchaudio._torchaudio'):
import atexit import atexit
init_sox_effects() init_sox_effects()
atexit.register(shutdown_sox_effects) atexit.register(shutdown_sox_effects)
__all__ = ['init_sox_effects', 'shutdown_sox_effects', 'effect_names', 'apply_effects_tensor', 'apply_effects_file']
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment