add __all__

audio/audiotools/core/effects.py

@@ -397,11 +397,6 @@ class EffectMixin:
         return self.convolve(other)
 
 
-import paddle
-import typing
-import numpy as np
-
-
 class ImpulseResponseMixin:
     """These functions are generally only used with AudioSignals that are derived
     from impulse responses, not other sources like music or speech. These methods

audio/audiotools/core/util.py

@@ -30,7 +30,28 @@ from flatten_dict import unflatten
 from .audio_signal import AudioSignal
 from paddlespeech.utils import satisfy_paddle_version
 
-# from ..data.preprocess import create_csv
+__all__ = [
+    "exp_compat",
+    "bool_index_compat",
+    "bool_setitem_compat",
+    "Info",
+    "info",
+    "ensure_tensor",
+    "random_state",
+    "seed",
+    "find_audio",
+    "read_sources",
+    "choose_from_list_of_lists",
+    "chdir",
+    "move_to_device",
+    "prepare_batch",
+    "sample_from_dist",
+    "format_figure",
+    "default_collate",
+    "collate",
+    "hz_to_bin",
+    "generate_chord_dataset",
+]
 
 
 def exp_compat(x):

audio/audiotools/data/datasets.py

@@ -12,6 +12,11 @@ from paddle.io import SequenceSampler
 from ..core import AudioSignal
 from ..core import util
 
+__all__ = [
+    "AudioLoader", "AudioDataset", "ConcatDataset",
+    "ResumableDistributedSampler", "ResumableSequentialSampler"
+]
+
 
 class AudioLoader:
     """Loads audio endlessly from a list of audio sources

audio/audiotools/data/transforms.py

@@ -15,6 +15,28 @@ from ..core import util
 from .datasets import AudioLoader
 from paddlespeech.utils import satisfy_paddle_version
 
+__all__ = [
+    "Identity",
+    "SpectralTransform",
+    "Compose",
+    "Choose",
+    "Repeat",
+    "RepeatUpTo",
+    "ClippingDistortion",
+    "Equalizer",
+    "BackgroundNoise",
+    "RoomImpulseResponse",
+    "VolumeNorm",
+    "GlobalVolumeNorm",
+    "Silence",
+    "LowPass",
+    "HighPass",
+    "FrequencyMask",
+    "TimeMask",
+    "Smoothing",
+    "FrequencyNoise",
+]
+
 
 class BaseTransform:
     """This is the base class for all transforms that are implemented