From 7a67f8f89701daeee3dd4b444bc44337f75fcca6 Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Tue, 8 Oct 2024 19:28:34 +0200 Subject: [PATCH] as func --- src/torchmetrics/functional/audio/pesq.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/torchmetrics/functional/audio/pesq.py b/src/torchmetrics/functional/audio/pesq.py index 530b0c14c79..ef0e4e81701 100644 --- a/src/torchmetrics/functional/audio/pesq.py +++ b/src/torchmetrics/functional/audio/pesq.py @@ -83,7 +83,10 @@ def perceptual_evaluation_speech_quality( ) import pesq as pesq_backend - _filter_error_msg = np.vectorize(lambda x: np.issubdtype(type(x), np.number)) + def _issubtype_number(x) -> bool: + return np.issubdtype(type(x), np.number) + + _filter_error_msg = np.vectorize(_issubtype_number) if fs not in (8000, 16000): raise ValueError(f"Expected argument `fs` to either be 8000 or 16000 but got {fs}")