diff --git a/src/functions.rs b/src/functions.rs index ecbaccbb3..09cdee619 100644 --- a/src/functions.rs +++ b/src/functions.rs @@ -183,6 +183,7 @@ fn list_indexof(array: PyExpr, element: PyExpr, index: Option) -> PyExpr { } #[pyfunction] +#[pyo3(signature = (array, begin, end, stride = None))] fn array_slice(array: PyExpr, begin: PyExpr, end: PyExpr, stride: Option) -> PyExpr { datafusion_functions_array::expr_fn::array_slice( array.into(), @@ -194,6 +195,7 @@ fn array_slice(array: PyExpr, begin: PyExpr, end: PyExpr, stride: Option } #[pyfunction] +#[pyo3(signature = (array, begin, end, stride = None))] fn list_slice(array: PyExpr, begin: PyExpr, end: PyExpr, stride: Option) -> PyExpr { // alias of array_slice array_slice(array, begin, end, stride)