From 003bab91f95ab0b40df32ba25d080ee9d753e74a Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 15 Nov 2022 21:39:47 +0530
Subject: [PATCH] tests

---
 tests/models/demucs_test.py   | 4 +++-
 tests/models/test_dccrn.py    | 4 +++-
 tests/models/test_waveunet.py | 4 +++-
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/tests/models/demucs_test.py b/tests/models/demucs_test.py
index e1203b7..0472cc2 100644
--- a/tests/models/demucs_test.py
+++ b/tests/models/demucs_test.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset
 
 
diff --git a/tests/models/test_dccrn.py b/tests/models/test_dccrn.py
index bc2a039..b309cb7 100644
--- a/tests/models/test_dccrn.py
+++ b/tests/models/test_dccrn.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset
 
 
diff --git a/tests/models/test_waveunet.py b/tests/models/test_waveunet.py
index bc250d1..ca0af0a 100644
--- a/tests/models/test_waveunet.py
+++ b/tests/models/test_waveunet.py
@@ -15,7 +15,9 @@ def vctk_dataset():
         test_clean="clean_testset_wav",
         test_noisy="noisy_testset_wav",
     )
-    dataset = MayaDataset(name="vctk", root_dir=root_dir, files=files)
+    dataset = MayaDataset(
+        name="vctk", root_dir=root_dir, files=files, sampling_rate=16000
+    )
     return dataset