Skip to content

Commit 7ce5941

Browse files
authored
🐛 signal: accept numbers as args to width and plateau_size for find_peaks() (#519)
1 parent 1c97aac commit 7ce5941

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

scipy-stubs/signal/_peak_finding.pyi

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,15 +69,17 @@ def peak_widths(
6969
#
7070
def find_peaks(
7171
x: onp.ToArray1D,
72-
height: onp.ToFloatND | Sequence[onp.ToFloat | None] | onp.ToFloat | None = None,
73-
threshold: onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
72+
height: onp.ToFloat | onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
73+
threshold: onp.ToInt | onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
7474
distance: onp.ToFloat | None = None,
7575
prominence: onp.ToFloat | onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
76-
width: onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
76+
width: onp.ToFloat | onp.ToFloatND | Sequence[onp.ToFloat | None] | None = None,
7777
wlen: onp.ToFloat | None = None,
7878
rel_height: onp.ToFloat = 0.5,
79-
plateau_size: onp.ToIntND | Sequence[onp.ToInt | None] | None = None,
79+
plateau_size: onp.ToInt | onp.ToIntND | Sequence[onp.ToInt | None] | None = None,
8080
) -> tuple[_Int1D, _FindPeaksResultsDict]: ...
81+
82+
#
8183
def find_peaks_cwt(
8284
vector: onp.Array,
8385
widths: onp.ToFloat | onp.ToFloatND,

0 commit comments

Comments
 (0)