diff --git a/ballista/core/src/execution_plans/distributed_query.rs b/ballista/core/src/execution_plans/distributed_query.rs index 2f12a6740..4e7a1c2be 100644 --- a/ballista/core/src/execution_plans/distributed_query.rs +++ b/ballista/core/src/execution_plans/distributed_query.rs @@ -21,7 +21,7 @@ use crate::serde::protobuf::execute_query_params::OptionalSessionId; use crate::serde::protobuf::{ execute_query_params::Query, execute_query_result, job_status, scheduler_grpc_client::SchedulerGrpcClient, ExecuteQueryParams, GetJobStatusParams, - GetJobStatusResult, PartitionLocation, + GetJobStatusResult, KeyValuePair, PartitionLocation, }; use crate::utils::create_grpc_client_connection; use datafusion::arrow::datatypes::SchemaRef; @@ -208,7 +208,15 @@ impl ExecutionPlan for DistributedQueryExec { let query = ExecuteQueryParams { query: Some(Query::LogicalPlan(buf)), - settings: vec![], + settings: self + .config + .settings() + .iter() + .map(|(k, v)| KeyValuePair { + key: k.to_owned(), + value: v.to_owned(), + }) + .collect::>(), optional_session_id: Some(OptionalSessionId::SessionId( self.session_id.clone(), )),