From aea4c51230d63ac4828187005801576db3b87835 Mon Sep 17 00:00:00 2001 From: saadtajwar Date: Fri, 31 Jul 2026 21:47:09 -0400 Subject: [PATCH] fix: rows_to_array cleanup for expecting single field --- .../aggregates/group_values/multi_group_by/row_backed.rs | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/row_backed.rs b/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/row_backed.rs index 29beb3bd66229..31735559cdb42 100644 --- a/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/row_backed.rs +++ b/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/row_backed.rs @@ -213,8 +213,13 @@ impl RowsGroupColumn { .row_converter .convert_rows(rows) .expect("row conversion during emit"); - debug_assert_eq!(arrays.len(), 1, "single-field row converter"); - let array = arrays.swap_remove(0); + assert_eq!( + arrays.len(), + 1, + "Single field row converter must produce exactly one array, actual length is {}", + arrays.len() + ); + let array = arrays.pop().unwrap(); encode_array_if_necessary(&array, &self.output_type) .expect("dictionary re-encode during emit") }