increase test tolerance for cWMM and CBMM

fgnt · Sep 15, 2020 · aa9e7e1 · aa9e7e1
1 parent 92300c1
commit aa9e7e1
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_distribution/test_spatial_mm.py b/tests/test_distribution/test_spatial_mm.py
@@ -122,10 +122,10 @@ def test_cwgmm():
     scores = trainer_on_simulated_speech_data(CWMMTrainer)
     np.testing.assert_allclose(
         scores['invasive_sxr_sdr'], [17.47441, 20.946751],
-        err_msg=str(scores))
+        err_msg=str(scores), rtol=1e-6)
     np.testing.assert_allclose(
         scores['mir_eval_sxr_sdr'], [9.675817, 13.557824],
-        err_msg=str(scores))
+        err_msg=str(scores), rtol=1e-6)
 
     np.random.seed(0)
     scores = trainer_on_simulated_speech_data(CWMMTrainer, reverberation=True)
@@ -134,7 +134,7 @@ def test_cwgmm():
         err_msg=str(scores), rtol=1e-6)
     np.testing.assert_allclose(
         scores['mir_eval_sxr_sdr'], [2.50231548, 3.08808406],
-        err_msg=str(scores))
+        err_msg=str(scores), rtol=1e-6)
 
 
 def test_cbgmm():
@@ -145,5 +145,5 @@ def test_cbgmm():
         scores['invasive_sxr_sdr'], [-0.51113, -3.246796],
         err_msg=str(scores), rtol=1e-5)
     np.testing.assert_allclose(
-        scores['mir_eval_sxr_sdr'], [-3.253749, -10.07552],
-        err_msg=str(scores))
+        scores['mir_eval_sxr_sdr'], [-3.253749, -10.075523],
+        err_msg=str(scores), rtol=1e-6)