chore

apache · Feb 6, 2025 · fa91ea3 · fa91ea3
1 parent f6934d1
commit fa91ea3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datafusion/physical-plan/src/repartition/on_demand_repartition.rs b/datafusion/physical-plan/src/repartition/on_demand_repartition.rs
@@ -620,7 +620,7 @@ impl Stream for OnDemandPerPartitionStream {
         cx: &mut Context<'_>,
     ) -> Poll<Option<Self::Item>> {
         if !self.is_requested && !self.sender.is_closed() {
-            self.sender.send_blocking(self.partition).map_err(|e| {
+            self.sender.try_send(self.partition).map_err(|e| {
                 internal_datafusion_err!(
                     "Error sending partition number to the receiver for partition {}: {}",
                     self.partition,
@@ -693,7 +693,7 @@ impl Stream for OnDemandRepartitionStream {
         loop {
             // Send partition number to input partitions
             if !self.is_requested && !self.sender.is_closed() {
-                self.sender.send_blocking(self.partition).map_err(|e| {
+                self.sender.try_send(self.partition).map_err(|e| {
                     internal_datafusion_err!(
                         "Error sending partition number to the receiver for partition {}: {}",
                         self.partition,