From 633310cd545b7c4bcf308b886fbaab79a523e0df Mon Sep 17 00:00:00 2001 From: Pavadol Yamsiri Date: Sun, 24 Nov 2024 16:36:53 +1100 Subject: [PATCH] `signal`: Move `gausspulse` overloads around to satisfy mypy This doesn't fix `stubtest` however but it does fix `typetest`. --- scipy-stubs/signal/_waveforms.pyi | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/scipy-stubs/signal/_waveforms.pyi b/scipy-stubs/signal/_waveforms.pyi index 8e0c03f6..ac1483aa 100644 --- a/scipy-stubs/signal/_waveforms.pyi +++ b/scipy-stubs/signal/_waveforms.pyi @@ -73,6 +73,18 @@ def unit_impulse( dtype: _DTypeLike[_SCT], ) -> npt.NDArray[_SCT]: ... +# Overloads for gausspulse when `t` is `"cutoff"` +@overload # retquad: False = ..., retenv: False = ... +def gausspulse( + t: Literal["cutoff"], + fc: onp.ToFloat = 1000, + bw: onp.ToFloat = 0.5, + bwr: onp.ToFloat = -6, + tpr: onp.ToFloat = -60, + retquad: op.CanBool = False, + retenv: op.CanBool = False, +) -> np.float64: ... + # Overloads for gausspulse when `t` is scalar @overload # retquad: False = ..., retenv: False = ... def gausspulse( @@ -222,15 +234,3 @@ def gausspulse( retquad: _Truthy, retenv: _Truthy, ) -> tuple[_Array_f8, _Array_f8, _Array_f8]: ... - -# Overloads for gausspulse when `t` is `"cutoff"` -@overload # retquad: False = ..., retenv: False = ... -def gausspulse( - t: Literal["cutoff"], - fc: onp.ToFloat = 1000, - bw: onp.ToFloat = 0.5, - bwr: onp.ToFloat = -6, - tpr: onp.ToFloat = -60, - retquad: op.CanBool = False, - retenv: op.CanBool = False, -) -> np.float64: ...