diff --git a/torchaudio/__init__.py b/torchaudio/__init__.py index 5fec11d924..35f5fd41d5 100644 --- a/torchaudio/__init__.py +++ b/torchaudio/__init__.py @@ -1,5 +1,5 @@ -from . import extension -from torchaudio._internal import module_utils as _mod_utils +from . import extension # noqa: F401 +from torchaudio._internal import module_utils as _mod_utils # noqa: F401 from torchaudio import ( compliance, datasets, @@ -31,8 +31,23 @@ 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'] +__all__ = [ + '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/compliance/__init__.py b/torchaudio/compliance/__init__.py index e520566c8a..795065dc8d 100644 --- a/torchaudio/compliance/__init__.py +++ b/torchaudio/compliance/__init__.py @@ -1,3 +1,5 @@ from . import kaldi -__all__ = ['kaldi'] +__all__ = [ + 'kaldi', +] diff --git a/torchaudio/datasets/__init__.py b/torchaudio/datasets/__init__.py index 5e5b9c1118..a83bf33c82 100644 --- a/torchaudio/datasets/__init__.py +++ b/torchaudio/datasets/__init__.py @@ -10,7 +10,7 @@ from .libritts import LIBRITTS from .tedlium import TEDLIUM -__all__ = ( +__all__ = [ "COMMONVOICE", "LIBRISPEECH", "SPEECHCOMMANDS", @@ -24,4 +24,4 @@ "diskcache_iterator", "bg_iterator", "TEDLIUM", -) +] diff --git a/torchaudio/functional/__init__.py b/torchaudio/functional/__init__.py index b63aa14949..d59400e82e 100644 --- a/torchaudio/functional/__init__.py +++ b/torchaudio/functional/__init__.py @@ -42,10 +42,44 @@ 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'] +__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 76e91ff166..5b134345af 100644 --- a/torchaudio/models/__init__.py +++ b/torchaudio/models/__init__.py @@ -2,4 +2,8 @@ from .wavernn import WaveRNN from .conv_tasnet import ConvTasNet -__all__ = ['Wav2Letter', 'WaveRNN', 'ConvTasNet'] +__all__ = [ + 'Wav2Letter', + 'WaveRNN', + 'ConvTasNet', +] diff --git a/torchaudio/models/wav2letter.py b/torchaudio/models/wav2letter.py index acd4dfe4b3..20a665a784 100644 --- a/torchaudio/models/wav2letter.py +++ b/torchaudio/models/wav2letter.py @@ -1,7 +1,9 @@ from torch import Tensor from torch import nn -__all__ = ["Wav2Letter"] +__all__ = [ + "Wav2Letter", +] class Wav2Letter(nn.Module): diff --git a/torchaudio/models/wavernn.py b/torchaudio/models/wavernn.py index 8fbd111a72..eedbd3c589 100644 --- a/torchaudio/models/wavernn.py +++ b/torchaudio/models/wavernn.py @@ -4,7 +4,13 @@ from torch import Tensor from torch import nn -__all__ = ["ResBlock", "MelResNet", "Stretch2d", "UpsampleNetwork", "WaveRNN"] +__all__ = [ + "ResBlock", + "MelResNet", + "Stretch2d", + "UpsampleNetwork", + "WaveRNN", +] class ResBlock(nn.Module): diff --git a/torchaudio/prototype/transducer.py b/torchaudio/prototype/transducer.py index d08ab6ea17..843cf5f9a4 100644 --- a/torchaudio/prototype/transducer.py +++ b/torchaudio/prototype/transducer.py @@ -5,7 +5,10 @@ module_utils as _mod_utils, ) -__all__ = ["rnnt_loss", "RNNTLoss"] +__all__ = [ + "rnnt_loss", + "RNNTLoss", +] class _RNNT(Function): diff --git a/torchaudio/sox_effects/__init__.py b/torchaudio/sox_effects/__init__.py index c9f70f8052..b3e2708ad1 100644 --- a/torchaudio/sox_effects/__init__.py +++ b/torchaudio/sox_effects/__init__.py @@ -13,4 +13,10 @@ init_sox_effects() atexit.register(shutdown_sox_effects) -__all__ = ['init_sox_effects', 'shutdown_sox_effects', 'effect_names', 'apply_effects_tensor', 'apply_effects_file'] +__all__ = [ + 'init_sox_effects', + 'shutdown_sox_effects', + 'effect_names', + 'apply_effects_tensor', + 'apply_effects_file', +]