update tests

Signed-off-by: Jason <[email protected]>

Author: blisc

tests/collections/audio/test_audio_models_flow_matching.py

@@ -79,7 +79,7 @@ def flow_matching_base_config(request):
         'time_max': flow['time_max'],
     }
 
-    loss = {'_target_': 'nemo.collections.audio.losses.MSELoss', 'ndim': 4}
+    loss = {'_target_': 'nemo.collections.audio.losses.audio.MSELoss', 'ndim': 4}
 
     estimator = {
         '_target_': 'nemo.collections.audio.parts.submodules.transformerunet.SpectrogramTransformerUNet',

tests/collections/audio/test_audio_models_mask.py

@@ -111,7 +111,7 @@ def mask_model_rnn_params():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.SDRLoss',
+        '_target_': 'nemo.collections.audio.losses.audio.SDRLoss',
         'scale_invariant': True,
     }
 
@@ -212,7 +212,7 @@ def mask_model_flexarray():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.SDRLoss',
+        '_target_': 'nemo.collections.audio.losses.audio.SDRLoss',
         'scale_invariant': True,
     }
 

tests/collections/audio/test_audio_models_predictive.py

@@ -111,7 +111,7 @@ def predictive_model_ncsn():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(
@@ -183,7 +183,7 @@ def predictive_model_conformer():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(
@@ -255,7 +255,7 @@ def predictive_model_streaming_conformer():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(
@@ -318,7 +318,7 @@ def predictive_model_transformer_unet_params_base():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(
@@ -384,7 +384,7 @@ def predictive_model_conformer_unet():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(
@@ -456,7 +456,7 @@ def predictive_model_streaming_conformer_unet():
     }
 
     loss = {
-        '_target_': 'nemo.collections.audio.losses.MSELoss',  # computed in the time domain
+        '_target_': 'nemo.collections.audio.losses.audio.MSELoss',  # computed in the time domain
     }
 
     model_config = DictConfig(

tests/collections/audio/test_audio_models_schroedinger_bridge.py

@@ -112,9 +112,9 @@ def schroedinger_bridge_model_ncsn_params():
         'pad_dimension_to': 0,  # no padding in the frequency dimension
     }
 
-    loss_encoded = {'_target_': 'nemo.collections.audio.losses.MSELoss', 'ndim': 4}  # computed in the time domain
+    loss_encoded = {'_target_': 'nemo.collections.audio.losses.audio.MSELoss', 'ndim': 4}  # computed in the time domain
 
-    loss_time = {'_target_': 'nemo.collections.audio.losses.MAELoss'}
+    loss_time = {'_target_': 'nemo.collections.audio.losses.audio.MAELoss'}
 
     noise_schedule = {
         '_target_': 'nemo.collections.audio.parts.submodules.schroedinger_bridge.SBNoiseScheduleVE',

tests/collections/audio/test_audio_models_score_based.py

@@ -87,7 +87,7 @@ def score_based_base_config():
         'snr': 0.5,
     }
 
-    loss = {'_target_': 'nemo.collections.audio.losses.MSELoss', 'ndim': 4}
+    loss = {'_target_': 'nemo.collections.audio.losses.audio.MSELoss', 'ndim': 4}
 
     trainer = {
         'max_epochs': -1,