Skip to content

Commit 1679eba

Browse files
committed
1 parent 501b2ee commit 1679eba

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pyDF/src/lib.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ struct DF {
2020
#[allow(clippy::upper_case_acronyms)]
2121
impl 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

Comments
 (0)