Fix type in two doctests

fgnt · Oct 17, 2024 · 13a14c2 · 13a14c2
1 parent a13e67a
commit 13a14c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pb_bss/evaluation/module_srmr.py b/pb_bss/evaluation/module_srmr.py
@@ -18,7 +18,7 @@ def srmr(signal, sample_rate: int = 16000, n_cochlear_filters: int = 23, low_fre
         >>> a = pb.testing.testfile_fetcher.get_file_path('speech_bab_0dB.wav')
         >>> a = pb.io.load_audio(a)
         >>> srmr(a, 16000)  # doctest: +ELLIPSIS
-        1.8561615800...
+        np.float64(1.8561615800...)
         >>> srmr([a, a], 16000)
         array([1.85616158, 1.85616158])
     """

diff --git a/pb_bss/evaluation/sxr_module.py b/pb_bss/evaluation/sxr_module.py
@@ -41,7 +41,7 @@ def get_snr(X, N, *, axis=None, keepdims=False):
         SNR of time signals or STFT signals in dB.
 
     >>> get_snr([1, 2, 3], [1, 2, 3])
-    0.0
+    np.float64(0.0)
     """
     power_X = get_variance_for_zero_mean_signal(X, axis=axis, keepdims=keepdims)
     power_N = get_variance_for_zero_mean_signal(N, axis=axis, keepdims=keepdims)