From 13a14c263535ea5e864b9a89f48aaa2c11c83679 Mon Sep 17 00:00:00 2001 From: sibange Date: Thu, 17 Oct 2024 11:22:37 +0200 Subject: [PATCH] Fix type in two doctests --- pb_bss/evaluation/module_srmr.py | 2 +- pb_bss/evaluation/sxr_module.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pb_bss/evaluation/module_srmr.py b/pb_bss/evaluation/module_srmr.py index 4d16e4a..0ee6751 100644 --- 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 index be95863..d9da950 100644 --- 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)