diff --git a/crates/polars-stream/src/nodes/joins/equi_join.rs b/crates/polars-stream/src/nodes/joins/equi_join.rs index c0c072d51d8b..524b3814ca46 100644 --- a/crates/polars-stream/src/nodes/joins/equi_join.rs +++ b/crates/polars-stream/src/nodes/joins/equi_join.rs @@ -516,6 +516,13 @@ impl ProbeState { } } +impl Drop for ProbeState { + fn drop(&mut self) { + // Parallel drop as the state might be quite big. + self.table_per_partition.par_drain(..).for_each(drop); + } +} + struct EmitUnmatchedState { partitions: Vec, active_partition_idx: usize,