fixing speech enhancing test

sensein · Sep 19, 2024 · 3548b61 · 3548b61
1 parent e3d5acc
commit 3548b61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tests/audio/tasks/speech_enhancement_test.py b/src/tests/audio/tasks/speech_enhancement_test.py
@@ -110,6 +110,6 @@ def test_enhance_audios_with_extreme_amplitude(audio_with_extreme_amplitude: Aud
 def test_model_caching(resampled_mono_audio_sample: Audio) -> None:
     """Test model caching by enhancing audios with the same model multiple times."""
     SpeechBrainEnhancer.enhance_audios_with_speechbrain(audios=[resampled_mono_audio_sample])
-    assert len(SpeechBrainEnhancer._models) == 1
+    assert len(list(SpeechBrainEnhancer._models.keys())) == 1
     SpeechBrainEnhancer.enhance_audios_with_speechbrain(audios=[resampled_mono_audio_sample])
-    assert len(SpeechBrainEnhancer._models) == 1
+    assert len(list(SpeechBrainEnhancer._models.keys())) == 1