@@ -20,6 +20,7 @@ struct DF {
2020#[ allow( clippy:: upper_case_acronyms) ]
2121impl DF {
2222 #[ new]
23+ #[ pyo3( signature = ( sr, fft_size, hop_size, nb_bands=None , min_nb_erb_freqs=None ) ) ]
2324 fn new (
2425 sr : usize ,
2526 fft_size : usize ,
@@ -38,6 +39,7 @@ impl DF {
3839 }
3940 }
4041
42+ #[ pyo3( signature = ( input, reset=None ) ) ]
4143 fn analysis < ' py > (
4244 & mut self ,
4345 py : Python < ' py > ,
@@ -71,6 +73,7 @@ impl DF {
7173 Ok ( output. into_pyarray ( py) )
7274 }
7375
76+ #[ pyo3( signature = ( input, reset=None ) ) ]
7477 fn synthesis < ' py > (
7578 & mut self ,
7679 py : Python < ' py > ,
@@ -141,6 +144,7 @@ fn libdf(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
141144
142145 #[ pyfn( m) ]
143146 #[ pyo3( name = "erb" ) ]
147+ #[ pyo3( signature = ( input, erb_fb, db=None ) ) ]
144148 fn erb < ' py > (
145149 py : Python < ' py > ,
146150 input : PyReadonlyArrayDyn < Complex32 > ,
@@ -251,6 +255,7 @@ fn libdf(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
251255
252256 #[ pyfn( m) ]
253257 #[ pyo3( name = "erb_norm" ) ]
258+ #[ pyo3( signature = ( erb, alpha, state=None ) ) ]
254259 fn erb_norm < ' py > (
255260 py : Python < ' py > ,
256261 erb : PyReadonlyArray3 < f32 > ,
@@ -275,6 +280,7 @@ fn libdf(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
275280
276281 #[ pyfn( m) ]
277282 #[ pyo3( name = "unit_norm" ) ]
283+ #[ pyo3( signature = ( spec, alpha, state=None ) ) ]
278284 fn unit_norm < ' py > (
279285 py : Python < ' py > ,
280286 spec : PyReadonlyArray3 < Complex32 > ,
0 commit comments