Remove noise_type argument

analogdevicesinc · Jul 2, 2024 · 68c5b73 · 68c5b73
1 parent 1e1305e
commit 68c5b73
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/datasets/kws20.py b/datasets/kws20.py
@@ -996,7 +996,6 @@ def KWS_35_get_unquantized_datasets(data, load_train=True, load_test=True):
 
 def KWS_20_msnoise_mixed_get_datasets(data, load_train=True, load_test=True,
                                       apply_prob=0.8, snr_range=(-5, 10),
-                                      noise_type=MSnoise.class_dict,
                                       desired_probs=None):
     """
     Returns the KWS dataset mixed with MSnoise dataset. Only training set will be mixed
@@ -1010,6 +1009,8 @@ def KWS_20_msnoise_mixed_get_datasets(data, load_train=True, load_test=True,
     noise_type --> All noise types in the noise dataset.
     """
 
+    noise_type = MSnoise.class_dict
+
     if len(snr_range) > 1:
         snr_range = range(snr_range[0], snr_range[1])
     else:
@@ -1052,14 +1053,15 @@ def KWS_12_benchmark_get_datasets(data, load_train=True, load_test=True):
 
 def MixedKWS_20_get_datasets_10dB(data, load_train=True, load_test=True,
                                   apply_prob=1, snr_range=tuple([10]),
-                                  noise_type=MSnoise.class_dict,
                                   desired_probs=None):
     """
     Returns the mixed KWS dataset with MSnoise dataset under 10 dB SNR using signalmixer
     data loader. All of the training and test data will be augmented with
     additional noise.
     """
 
+    noise_type = MSnoise.class_dict
+
     if len(snr_range) > 1:
         snr_range = range(snr_range[0], snr_range[1])
     else: