Skip to content

Commit 2455aab

Browse files
authored
fix: date_trunc bench broken by #15049 (#15169)
Closes #15167
1 parent 359735a commit 2455aab

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

datafusion/functions/benches/date_trunc.rs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,14 @@ fn criterion_benchmark(c: &mut Criterion) {
4646
ColumnarValue::Scalar(ScalarValue::Utf8(Some("minute".to_string())));
4747
let timestamps = ColumnarValue::Array(timestamps_array);
4848
let udf = date_trunc();
49-
let return_type = &udf.return_type(&[timestamps.data_type()]).unwrap();
49+
let args = vec![precision, timestamps];
50+
let return_type = &udf
51+
.return_type(&args.iter().map(|arg| arg.data_type()).collect::<Vec<_>>())
52+
.unwrap();
5053
b.iter(|| {
5154
black_box(
5255
udf.invoke_with_args(ScalarFunctionArgs {
53-
args: vec![precision.clone(), timestamps.clone()],
56+
args: args.clone(),
5457
number_rows: batch_len,
5558
return_type,
5659
})

0 commit comments

Comments
 (0)