From 6415906a545fd3a1749968aa027088e75992228d Mon Sep 17 00:00:00 2001
From: drryanhuang <zihaohuang@aliyun.com>
Date: Thu, 2 Jan 2025 04:32:56 +0000
Subject: [PATCH] add __all__

---
 audio/audiotools/core/effects.py    |  5 -----
 audio/audiotools/core/util.py       | 23 ++++++++++++++++++++++-
 audio/audiotools/data/datasets.py   |  5 +++++
 audio/audiotools/data/transforms.py | 22 ++++++++++++++++++++++
 4 files changed, 49 insertions(+), 6 deletions(-)

diff --git a/audio/audiotools/core/effects.py b/audio/audiotools/core/effects.py
index cda0b4f2e..5e5ac5535 100644
--- a/audio/audiotools/core/effects.py
+++ b/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
diff --git a/audio/audiotools/core/util.py b/audio/audiotools/core/util.py
index 0881dd0b4..1e97e97aa 100644
--- a/audio/audiotools/core/util.py
+++ b/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):
diff --git a/audio/audiotools/data/datasets.py b/audio/audiotools/data/datasets.py
index 61764c0bf..a4f1cd3fe 100644
--- a/audio/audiotools/data/datasets.py
+++ b/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
diff --git a/audio/audiotools/data/transforms.py b/audio/audiotools/data/transforms.py
index 102b46c79..701ceefd9 100644
--- a/audio/audiotools/data/transforms.py
+++ b/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