chore

Author: Weijun-H

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,