From 8a50671ae1b1728d8c0b2e2313b0b6aef7f8ffaa Mon Sep 17 00:00:00 2001 From: pytorchbot Date: Sat, 19 Oct 2024 11:34:54 +0000 Subject: [PATCH] 2024-10-19 nightly release (a95cfa82c25a45007b9fc57f0cd8911c5b8d2262) --- test/torchaudio_unittest/prototype/functional/dsp_utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/torchaudio_unittest/prototype/functional/dsp_utils.py b/test/torchaudio_unittest/prototype/functional/dsp_utils.py index fb0300a9d6..44c0cac3c3 100644 --- a/test/torchaudio_unittest/prototype/functional/dsp_utils.py +++ b/test/torchaudio_unittest/prototype/functional/dsp_utils.py @@ -1,4 +1,5 @@ import numpy as np +import numpy.typing as npt def oscillator_bank( @@ -43,8 +44,8 @@ def freq_ir(magnitudes): def exp_sigmoid( - input: np.ndarray, exponent: float = 10.0, max_value: float = 2.0, threshold: float = 1e-7 -) -> np.ndarray: + input: npt.NDArray, exponent: float = 10.0, max_value: float = 2.0, threshold: float = 1e-7 +) -> npt.NDArray: """Exponential Sigmoid pointwise nonlinearity (Numpy version). Implements the equation: ``max_value`` * sigmoid(``input``) ** (log(``exponent``)) + ``threshold``