Update datafusion/functions-aggregate/src/median.rs

Rachelint · alamb · Rachelint · commit 28866501dd31 · 2025-01-29T14:01:13.000+08:00
Co-authored-by: Andrew Lamb &lt;andrew@nerdnetworks.org&gt;
diff --git a/datafusion/functions-aggregate/src/median.rs b/datafusion/functions-aggregate/src/median.rs
@@ -468,6 +468,8 @@ impl<T: ArrowNumericType + Send> GroupsAccumulator for MedianGroupsAccumulator<T
             .iter()
             .map(|values| values.capacity() * size_of::<T>())
             .sum::<usize>()
+            // account for size of self.grou_values too
+            + self.group_values.capacity() * size_of::<Vec<T>>()
     }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -468,6 +468,8 @@ impl<T: ArrowNumericType + Send> GroupsAccumulator for MedianGroupsAccumulator<T`
`468`	`468`	`.iter()`
`469`	`469`	`.map(\|values\| values.capacity() * size_of::<T>())`
`470`	`470`	`.sum::<usize>()`
	`471`	`+ // account for size of self.grou_values too`
	`472`	`+ + self.group_values.capacity() * size_of::<Vec<T>>()`
`471`	`473`	`}`
`472`	`474`	`}`
`473`	`475`