diff --git a/datafusion/functions/src/encoding/inner.rs b/datafusion/functions/src/encoding/inner.rs index 68a6d1006052..51e8c6968866 100644 --- a/datafusion/functions/src/encoding/inner.rs +++ b/datafusion/functions/src/encoding/inner.rs @@ -103,12 +103,11 @@ impl ScalarUDFImpl for EncodeFunc { }) } - fn invoke_batch( + fn invoke_with_args( &self, - args: &[ColumnarValue], - _number_rows: usize, + args: datafusion_expr::ScalarFunctionArgs, ) -> Result { - encode(args) + encode(&args.args) } fn coerce_types(&self, arg_types: &[DataType]) -> Result> { @@ -183,12 +182,11 @@ impl ScalarUDFImpl for DecodeFunc { Ok(arg_types[0].to_owned()) } - fn invoke_batch( + fn invoke_with_args( &self, - args: &[ColumnarValue], - _number_rows: usize, + args: datafusion_expr::ScalarFunctionArgs, ) -> Result { - decode(args) + decode(&args.args) } fn coerce_types(&self, arg_types: &[DataType]) -> Result> {