diff --git a/.flake8 b/.flake8 index b4a5069f16..f344aec879 100644 --- a/.flake8 +++ b/.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 diff --git a/test/torchaudio_unittest/backend/soundfile/save_test.py b/test/torchaudio_unittest/backend/soundfile/save_test.py index 5c36e0d126..2f2741c303 100644 --- a/test/torchaudio_unittest/backend/soundfile/save_test.py +++ b/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 diff --git a/test/torchaudio_unittest/batch_consistency_test.py b/test/torchaudio_unittest/batch_consistency_test.py index 851fc9c86c..4644fe7dd3 100644 --- a/test/torchaudio_unittest/batch_consistency_test.py +++ b/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 diff --git a/test/torchaudio_unittest/common_utils/__init__.py b/test/torchaudio_unittest/common_utils/__init__.py index cf3e717116..8cc44c3529 100644 --- a/test/torchaudio_unittest/common_utils/__init__.py +++ b/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'] diff --git a/test/torchaudio_unittest/compliance_kaldi_test.py b/test/torchaudio_unittest/compliance_kaldi_test.py index 43e9865e2f..c7b4149819 100644 --- a/test/torchaudio_unittest/compliance_kaldi_test.py +++ b/test/torchaudio_unittest/compliance_kaldi_test.py @@ -1,6 +1,5 @@ import os import math -import unittest import torch import torchaudio diff --git a/test/torchaudio_unittest/datasets/datasets_test.py b/test/torchaudio_unittest/datasets/datasets_test.py index cd04c4d447..fda0804ea6 100644 --- a/test/torchaudio_unittest/datasets/datasets_test.py +++ b/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 ( diff --git a/test/torchaudio_unittest/kaldi_io_test.py b/test/torchaudio_unittest/kaldi_io_test.py index 2153b4cc02..4ceeeebbc5 100644 --- a/test/torchaudio_unittest/kaldi_io_test.py +++ b/test/torchaudio_unittest/kaldi_io_test.py @@ -1,5 +1,3 @@ -import unittest - import torch import torchaudio.kaldi_io as kio diff --git a/test/torchaudio_unittest/sox_compatibility_test.py b/test/torchaudio_unittest/sox_compatibility_test.py index ade98b2aeb..c68e529427 100644 --- a/test/torchaudio_unittest/sox_compatibility_test.py +++ b/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 diff --git a/test/torchaudio_unittest/sox_effect/smoke_test.py b/test/torchaudio_unittest/sox_effect/smoke_test.py index 2a8d9e7ff9..3b2b34d3ec 100644 --- a/test/torchaudio_unittest/sox_effect/smoke_test.py +++ b/test/torchaudio_unittest/sox_effect/smoke_test.py @@ -10,7 +10,6 @@ save_wav, ) from .common import ( - name_func, load_params, ) diff --git a/test/torchaudio_unittest/torchscript_consistency_impl.py b/test/torchaudio_unittest/torchscript_consistency_impl.py index ba8c93a1a7..492e1e4a92 100644 --- a/test/torchaudio_unittest/torchscript_consistency_impl.py +++ b/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 diff --git a/test/torchaudio_unittest/transforms_test.py b/test/torchaudio_unittest/transforms_test.py index 76628b1315..ab7b2ed981 100644 --- a/test/torchaudio_unittest/transforms_test.py +++ b/test/torchaudio_unittest/transforms_test.py @@ -1,5 +1,4 @@ import math -import unittest import torch import torchaudio diff --git a/torchaudio/__init__.py b/torchaudio/__init__.py index cbb2ba3370..5fec11d924 100644 --- a/torchaudio/__init__.py +++ b/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'] diff --git a/torchaudio/backend/__init__.py b/torchaudio/backend/__init__.py index 0812c09cf4..361935229f 100644 --- a/torchaudio/backend/__init__.py +++ b/torchaudio/backend/__init__.py @@ -1,3 +1,4 @@ +# flake8: noqa from . import utils from .utils import ( list_audio_backends, diff --git a/torchaudio/compliance/__init__.py b/torchaudio/compliance/__init__.py index ba354c371e..e520566c8a 100644 --- a/torchaudio/compliance/__init__.py +++ b/torchaudio/compliance/__init__.py @@ -1 +1,3 @@ from . import kaldi + +__all__ = ['kaldi'] diff --git a/torchaudio/datasets/gtzan.py b/torchaudio/datasets/gtzan.py index 4c91894be1..6c0b6246f3 100644 --- a/torchaudio/datasets/gtzan.py +++ b/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 diff --git a/torchaudio/datasets/ljspeech.py b/torchaudio/datasets/ljspeech.py index 99058dd83e..1a519242c3 100644 --- a/torchaudio/datasets/ljspeech.py +++ b/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 diff --git a/torchaudio/extension/extension.py b/torchaudio/extension/extension.py index b01ba13e39..5875f41023 100644 --- a/torchaudio/extension/extension.py +++ b/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 diff --git a/torchaudio/functional/__init__.py b/torchaudio/functional/__init__.py index 688708296b..b63aa14949 100644 --- a/torchaudio/functional/__init__.py +++ b/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'] diff --git a/torchaudio/models/__init__.py b/torchaudio/models/__init__.py index 55498fd7f6..76e91ff166 100644 --- a/torchaudio/models/__init__.py +++ b/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'] diff --git a/torchaudio/sox_effects/__init__.py b/torchaudio/sox_effects/__init__.py index 5baf406b10..c9f70f8052 100644 --- a/torchaudio/sox_effects/__init__.py +++ b/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'] diff --git a/torchaudio/sox_effects/sox_effects.py b/torchaudio/sox_effects/sox_effects.py index 4d1d0234d9..e183932788 100644 --- a/torchaudio/sox_effects/sox_effects.py +++ b/torchaudio/sox_effects/sox_effects.py @@ -1,5 +1,4 @@ import os -from pathlib import Path from typing import List, Tuple, Optional import torch diff --git a/torchaudio/transforms.py b/torchaudio/transforms.py index 820661ff30..af89d5b250 100644 --- a/torchaudio/transforms.py +++ b/torchaudio/transforms.py @@ -2,7 +2,6 @@ import math from typing import Callable, Optional -from warnings import warn import torch from torch import Tensor