Skip to content

Commit 08438a2

Browse files
add type hints to rauc
1 parent 891a2fc commit 08438a2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

elephant/signal_processing.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -851,7 +851,11 @@ def hilbert(signal: neo.AnalogSignal,
851851
return output / output.units
852852

853853

854-
def rauc(signal, baseline=None, bin_duration=None, t_start=None, t_stop=None):
854+
def rauc(signal: neo.AnalogSignal,
855+
baseline: pq.Quantity | str | None = None,
856+
bin_duration: pq.Quantity | None = None,
857+
t_start: pq.Quantity | None = None,
858+
t_stop: pq.Quantity | None = None) -> pq.Quantity | neo.AnalogSignal:
855859
"""
856860
Calculate the rectified area under the curve (RAUC) for a
857861
`neo.AnalogSignal`.

0 commit comments

Comments
 (0)