Fixing Tests

fgnt · Jul 7, 2022 · 02a97af · 02a97af
1 parent 11a3736
commit 02a97af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pb_bss/evaluation/wrapper.py b/pb_bss/evaluation/wrapper.py
@@ -294,7 +294,7 @@ def __init__(
              'mir_eval_sir': array([25.6896, 46.638 ]),
              'mir_eval_sar': array([7.3309, 7.3309]),
              'mir_eval_selection': array([0, 1]),
-             'srmr': array([125.2507, 126.1846])}
+             'srmr': array([203.3988, 203.3988])}
 
             # Obtain particular metric (e.g. pesq)
             >>> metrics.pesq

diff --git a/tests/test_evaluation/test_wrapper_values.py b/tests/test_evaluation/test_wrapper_values.py
@@ -89,7 +89,7 @@ def test_input_metrics():
                     [0.28424 , 0.345368, 0.279996]], rtol=1e-5)
         elif k == 'srmr':
             np.testing.assert_allclose(
-                v, [0.51143847, 0.49881274, 0.48087079], rtol=1e-6)
+                v, [0.51612031, 0.50214891, 0.48237807], rtol=1e-6)
         else:
             raise KeyError(k, v)
 
@@ -145,6 +145,6 @@ def test_output_metrics():
         elif k == 'mir_eval_selection':
             assert all(v == [0, 1])
         elif k == 'srmr':
-            np.testing.assert_allclose(v, [0.5504078, 0.50442512])
+            np.testing.assert_allclose(v, [0.54593548, 0.49966431])
         else:
             raise KeyError(k, v)