From ce794095468856470dca75a91a002ca5c59c44c1 Mon Sep 17 00:00:00 2001
From: iver56 <ij@aiascience.com>
Date: Mon, 12 Oct 2020 15:21:43 +0200
Subject: [PATCH] Remove imports that are not ready for prime time yet

---
 tests/test_background_noise.py    | 4 +++-
 tests/test_impulse_response.py    | 4 ++--
 torch_audiomentations/__init__.py | 7 -------
 3 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/tests/test_background_noise.py b/tests/test_background_noise.py
index bb61c8ae..498c3eb6 100644
--- a/tests/test_background_noise.py
+++ b/tests/test_background_noise.py
@@ -1,6 +1,8 @@
 import torch
 import unittest
-from torch_audiomentations import ApplyBackgroundNoise, load_audio, calculate_rms
+from torch_audiomentations.augmentations.background_noise import ApplyBackgroundNoise
+from torch_audiomentations.utils.dsp import calculate_rms
+from torch_audiomentations.utils.file import load_audio
 from .utils import TEST_FIXTURES_DIR
 
 
diff --git a/tests/test_impulse_response.py b/tests/test_impulse_response.py
index b6f09258..5ed8d6d0 100644
--- a/tests/test_impulse_response.py
+++ b/tests/test_impulse_response.py
@@ -2,9 +2,9 @@
 import unittest
 
 import torch
-from numpy.testing import assert_raises, assert_array_almost_equal
 
-from torch_audiomentations import ApplyImpulseResponse, load_audio
+from torch_audiomentations.augmentations.impulse_response import ApplyImpulseResponse
+from torch_audiomentations.utils.file import load_audio
 from .utils import TEST_FIXTURES_DIR
 
 
diff --git a/torch_audiomentations/__init__.py b/torch_audiomentations/__init__.py
index fa994270..a0a59271 100644
--- a/torch_audiomentations/__init__.py
+++ b/torch_audiomentations/__init__.py
@@ -1,11 +1,4 @@
-from .augmentations.background_noise import ApplyBackgroundNoise
-from .augmentations.impulse_response import ApplyImpulseResponse
 from .augmentations.gain import Gain
 from .augmentations.polarity_inversion import PolarityInversion
 
-# TODO: Revise these imports
-from .utils.convolution import convolve
-from .utils.dsp import calculate_rms, calculate_desired_noise_rms, resample_audio
-from .utils.file import find_audio_files, load_audio
-
 __version__ = "0.0.1"