fix(11523): remove to_root() for the error, since the immediate inner…

… child should be returning an OOM
influxdata · Jul 26, 2024 · 1b1223f · 1b1223f
1 parent a8383fa
commit 1b1223f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/datafusion/execution/src/memory_pool/pool.rs b/datafusion/execution/src/memory_pool/pool.rs
@@ -362,16 +362,16 @@ impl<I: MemoryPool> MemoryPool for TrackConsumersPool<I> {
     }
 
     fn try_grow(&self, reservation: &MemoryReservation, additional: usize) -> Result<()> {
-        self.inner.try_grow(reservation, additional).map_err(|e| {
-            match e.find_root() {
+        self.inner
+            .try_grow(reservation, additional)
+            .map_err(|e| match e {
                 DataFusionError::ResourcesExhausted(e) => {
                     DataFusionError::ResourcesExhausted(
                         e.to_owned() + ". " + &self.report_top(),
                     )
                 }
                 _ => e,
-            }
-        })?;
+            })?;
 
         self.tracked_consumers
             .lock()