Improve Flake Rules

.flake8

@@ -1,4 +1,4 @@
 [flake8]
 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

test/torchaudio_unittest/backend/soundfile/save_test.py

@@ -1,5 +1,4 @@
 import io
-import itertools
 from unittest.mock import patch
 
 from torchaudio._internal import module_utils as _mod_utils

test/torchaudio_unittest/batch_consistency_test.py

@@ -1,5 +1,4 @@
 """Test numerical consistency among single input and batched input."""
-import unittest
 import itertools
 from parameterized import parameterized
 

test/torchaudio_unittest/common_utils/__init__.py

@@ -27,4 +27,8 @@
 from .parameterized_utils import (
     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']

test/torchaudio_unittest/compliance_kaldi_test.py

@@ -1,6 +1,5 @@
 import os
 import math
-import unittest
 
 import torch
 import torchaudio

test/torchaudio_unittest/datasets/datasets_test.py

@@ -1,5 +1,3 @@
-from torchaudio.datasets.librispeech import LIBRISPEECH
-from torchaudio.datasets.speechcommands import SPEECHCOMMANDS
 from torchaudio.datasets.vctk import VCTK
 
 from torchaudio_unittest.common_utils import (

test/torchaudio_unittest/kaldi_io_test.py

@@ -1,5 +1,3 @@
-import unittest
-
 import torch
 import torchaudio.kaldi_io as kio
 

test/torchaudio_unittest/sox_compatibility_test.py

@@ -1,5 +1,3 @@
-import unittest
-
 import torch
 import torchaudio.functional as F
 import torchaudio.transforms as T

test/torchaudio_unittest/sox_effect/smoke_test.py

@@ -10,7 +10,6 @@
     save_wav,
 )
 from .common import (
-    name_func,
     load_params,
 )
 

test/torchaudio_unittest/torchscript_consistency_impl.py

@@ -1,5 +1,4 @@
 """Test suites for jit-ability and its numerical compatibility"""
-import unittest
 
 import torch
 import torchaudio.transforms as T

test/torchaudio_unittest/transforms_test.py

@@ -1,5 +1,4 @@
 import math
-import unittest
 
 import torch
 import torchaudio

torchaudio/__init__.py

@@ -30,3 +30,9 @@
     from .version import __version__, git_version  # noqa: F401
 except ImportError:
     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']

torchaudio/backend/__init__.py

@@ -1,3 +1,4 @@
+# flake8: noqa 
 from . import utils
 from .utils import (
     list_audio_backends,

torchaudio/compliance/__init__.py

@@ -1 +1,3 @@
 from . import kaldi
+
+__all__ = ['kaldi']

torchaudio/datasets/gtzan.py

@@ -1,7 +1,6 @@
 import os
-import warnings
 from pathlib import Path
-from typing import Any, Tuple, Optional, Union
+from typing import Tuple, Optional, Union
 
 import torchaudio
 from torch import Tensor

torchaudio/datasets/ljspeech.py

@@ -1,6 +1,6 @@
 import os
 import csv
-from typing import List, Tuple, Union
+from typing import Tuple, Union
 from pathlib import Path
 
 import torchaudio

torchaudio/extension/extension.py

@@ -1,6 +1,5 @@
 import warnings
 import importlib
-from collections import namedtuple
 
 import torch
 from torchaudio._internal import module_utils as _mod_utils

torchaudio/functional/__init__.py

@@ -41,3 +41,11 @@
     treble_biquad,
     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']

torchaudio/models/__init__.py

@@ -1,3 +1,5 @@
 from .wav2letter import Wav2Letter
 from .wavernn import WaveRNN
 from .conv_tasnet import ConvTasNet
+
+__all__ = ['Wav2Letter', 'WaveRNN', 'ConvTasNet']

torchaudio/sox_effects/__init__.py

@@ -12,3 +12,5 @@
     import atexit
     init_sox_effects()
     atexit.register(shutdown_sox_effects)
+
+__all__ = ['init_sox_effects', 'shutdown_sox_effects', 'effect_names', 'apply_effects_tensor', 'apply_effects_file']

torchaudio/sox_effects/sox_effects.py

@@ -1,5 +1,4 @@
 import os
-from pathlib import Path
 from typing import List, Tuple, Optional
 
 import torch

torchaudio/transforms.py

@@ -2,7 +2,6 @@
 
 import math
 from typing import Callable, Optional
-from warnings import warn
 
 import torch
 from torch import Tensor