From 2b89ce1776ba4f7971834652ee03f841580609a7 Mon Sep 17 00:00:00 2001 From: Andrew Gazelka Date: Tue, 19 Nov 2024 13:54:13 -0800 Subject: [PATCH] fix stream --- src/daft-connect/src/op/execute/root.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/daft-connect/src/op/execute/root.rs b/src/daft-connect/src/op/execute/root.rs index 951e39619f..66fc1f7ab8 100644 --- a/src/daft-connect/src/op/execute/root.rs +++ b/src/daft-connect/src/op/execute/root.rs @@ -4,7 +4,7 @@ use common_daft_config::DaftExecutionConfig; use futures::stream; use spark_connect::{ExecutePlanResponse, Relation}; use tokio_util::sync::CancellationToken; -use tonic::{codegen::tokio_stream::wrappers::UnboundedReceiverStream, Status}; +use tonic::{codegen::tokio_stream::wrappers::ReceiverStream, Status}; use crate::{ op::execute::{ExecuteStream, PlanIds}, @@ -61,7 +61,7 @@ impl Session { } }); - let stream = UnboundedReceiverStream::new(rx); + let stream = ReceiverStream::new(rx); let stream = stream .map_err(|e| Status::internal(format!("Error in Daft server: {e:?}")))