use futures::stream

apache · Jun 6, 2023 · fb17af8 · fb17af8
1 parent 70a3d57
commit fb17af8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/datafusion/core/src/physical_plan/stream.rs b/datafusion/core/src/physical_plan/stream.rs
@@ -193,10 +193,9 @@ impl RecordBatchReceiverStreamBuilder {
             .filter_map(|item| async move { item });
 
         // Merge the streams together so whichever is ready first
-        // produces the batch (since futures::stream:StreamExt is
-        // already in scope, need to call it explicitly)
+        // produces the batch
         let inner =
-            tokio_stream::StreamExt::merge(ReceiverStream::new(rx), check_stream).boxed();
+            futures::stream::select(ReceiverStream::new(rx), check_stream).boxed();
 
         Box::pin(RecordBatchReceiverStream { schema, inner })
     }