fix: update arrow_cast function to accept Expr type for data_type parameter

Author: kosiew

python/datafusion/functions.py

@@ -1100,9 +1100,9 @@ def arrow_typeof(arg: Expr) -> Expr:
     return Expr(f.arrow_typeof(arg.expr))
 
 
-def arrow_cast(expr: Expr, data_type: str) -> Expr:
+def arrow_cast(expr: Expr, data_type: Expr) -> Expr:
     """Casts an expression to a specified data type."""
-    return Expr(f.arrow_cast(expr.expr, data_type))
+    return Expr(f.arrow_cast(expr.expr, data_type.expr))
 
 
 def random() -> Expr:

src/functions.rs

@@ -563,7 +563,7 @@ expr_fn_vec!(r#struct); // Use raw identifier since struct is a keyword
 expr_fn_vec!(named_struct);
 expr_fn!(from_unixtime, unixtime);
 expr_fn!(arrow_typeof, arg_1);
-expr_fn!(arrow_cast, datatype);
+expr_fn!(arrow_cast, column datatype);
 expr_fn!(random);
 
 // Array Functions
@@ -856,6 +856,7 @@ pub(crate) fn init_module(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_wrapped(wrap_pyfunction!(range))?;
     m.add_wrapped(wrap_pyfunction!(array_agg))?;
     m.add_wrapped(wrap_pyfunction!(arrow_typeof))?;
+    m.add_wrapped(wrap_pyfunction!(arrow_cast))?;
     m.add_wrapped(wrap_pyfunction!(ascii))?;
     m.add_wrapped(wrap_pyfunction!(asin))?;
     m.add_wrapped(wrap_pyfunction!(asinh))?;