test

viirya · Oct 16, 2024 · 67f86ae · 67f86ae
1 parent 1e2c4ed
commit 67f86ae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/native/Cargo.lock b/native/Cargo.lock
diff --git a/native/core/src/execution/datafusion/shuffle_writer.rs b/native/core/src/execution/datafusion/shuffle_writer.rs
@@ -56,6 +56,7 @@ use datafusion::{
 };
 use datafusion_physical_expr::EquivalenceProperties;
 use futures::{lock::Mutex, Stream, StreamExt, TryFutureExt, TryStreamExt};
+use futures::executor::block_on;
 use itertools::Itertools;
 use simd_adler32::Adler32;
 use tokio::task;
@@ -1139,7 +1140,7 @@ async fn external_shuffle(
     );
 
     while let Some(batch) = input.next().await {
-        repartitioner.insert_batch(batch?).await?;
+        block_on(repartitioner.insert_batch(batch?))?;
     }
     repartitioner.shuffle_write().await
 }