From 9dbcef4a8f6e7a77c776db9e4b49ea59e268eaa5 Mon Sep 17 00:00:00 2001 From: Arttu Voutilainen Date: Tue, 11 Mar 2025 22:56:29 +0000 Subject: [PATCH] fix: date_trunc bench broken by #15049 Closes https://github.com/apache/datafusion/issues/15167 --- datafusion/functions/benches/date_trunc.rs | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/datafusion/functions/benches/date_trunc.rs b/datafusion/functions/benches/date_trunc.rs index b7efe7cc8d0a..e7e96fb7a9fa 100644 --- a/datafusion/functions/benches/date_trunc.rs +++ b/datafusion/functions/benches/date_trunc.rs @@ -46,11 +46,14 @@ fn criterion_benchmark(c: &mut Criterion) { ColumnarValue::Scalar(ScalarValue::Utf8(Some("minute".to_string()))); let timestamps = ColumnarValue::Array(timestamps_array); let udf = date_trunc(); - let return_type = &udf.return_type(&[timestamps.data_type()]).unwrap(); + let args = vec![precision, timestamps]; + let return_type = &udf + .return_type(&args.iter().map(|arg| arg.data_type()).collect::>()) + .unwrap(); b.iter(|| { black_box( udf.invoke_with_args(ScalarFunctionArgs { - args: vec![precision.clone(), timestamps.clone()], + args: args.clone(), number_rows: batch_len, return_type, })