Skip to content

Commit

Permalink
update concat and concat_ws to use datafusion_functions
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael-J-Ward committed May 13, 2024
1 parent f266cd3 commit 2be45eb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/functions.rs
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ fn digest(value: PyExpr, method: PyExpr) -> PyExpr {
#[pyo3(signature = (*args))]
fn concat(args: Vec<PyExpr>) -> PyResult<PyExpr> {
let args = args.into_iter().map(|e| e.expr).collect::<Vec<_>>();
Ok(datafusion_expr::concat(&args).into())
Ok(functions::string::expr_fn::concat(args).into())
}

/// Concatenates all but the first argument, with separators.
Expand All @@ -144,7 +144,7 @@ fn concat(args: Vec<PyExpr>) -> PyResult<PyExpr> {
#[pyo3(signature = (sep, *args))]
fn concat_ws(sep: String, args: Vec<PyExpr>) -> PyResult<PyExpr> {
let args = args.into_iter().map(|e| e.expr).collect::<Vec<_>>();
Ok(datafusion_expr::concat_ws(lit(sep), args).into())
Ok(functions::string::expr_fn::concat_ws(lit(sep), args).into())
}

/// Creates a new Sort Expr
Expand Down

0 comments on commit 2be45eb

Please sign in to comment.