From f3769a6179554978edfebb27dbc6869395e22910 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marko=20Milenkovi=C4=87?= Date: Fri, 8 Nov 2024 22:10:25 +0000 Subject: [PATCH] replace deprecations --- ballista/core/src/utils.rs | 8 ++++---- ballista/executor/src/executor_process.rs | 2 +- ballista/executor/src/standalone.rs | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/ballista/core/src/utils.rs b/ballista/core/src/utils.rs index d4c202254..0745f9d09 100644 --- a/ballista/core/src/utils.rs +++ b/ballista/core/src/utils.rs @@ -71,7 +71,7 @@ pub fn default_session_builder(config: SessionConfig) -> SessionState { .with_default_features() .with_config(config) .with_runtime_env(Arc::new( - RuntimeEnv::new(with_object_store_registry(RuntimeConfig::default())) + RuntimeEnv::try_new(with_object_store_registry(RuntimeConfig::default())) .unwrap(), )) .build() @@ -268,7 +268,7 @@ pub fn create_df_ctx_with_ballista_query_planner( .with_default_features() .with_config(session_config) .with_runtime_env(Arc::new( - RuntimeEnv::new(with_object_store_registry(RuntimeConfig::default())) + RuntimeEnv::try_new(with_object_store_registry(RuntimeConfig::default())) .unwrap(), )) .with_query_planner(planner) @@ -319,7 +319,7 @@ impl SessionStateExt for SessionState { .with_round_robin_repartition(false); let runtime_config = RuntimeConfig::default(); - let runtime_env = RuntimeEnv::new(runtime_config)?; + let runtime_env = RuntimeEnv::try_new(runtime_config)?; let session_state = SessionStateBuilder::new() .with_default_features() .with_config(session_config) @@ -709,7 +709,7 @@ mod test { use crate::utils::{LocalRun, SessionStateExt}; fn context() -> SessionContext { - let runtime_environment = RuntimeEnv::new(RuntimeConfig::new()).unwrap(); + let runtime_environment = RuntimeEnv::try_new(RuntimeConfig::new()).unwrap(); let session_config = SessionConfig::new().with_information_schema(true); diff --git a/ballista/executor/src/executor_process.rs b/ballista/executor/src/executor_process.rs index e6f034f46..8e2e2ebe8 100644 --- a/ballista/executor/src/executor_process.rs +++ b/ballista/executor/src/executor_process.rs @@ -203,7 +203,7 @@ pub async fn start_executor_process(opt: Arc) -> Result<( let wd = work_dir.clone(); let runtime_producer: RuntimeProducer = Arc::new(move |_| { let config = RuntimeConfig::new().with_temp_file_path(wd.clone()); - Ok(Arc::new(RuntimeEnv::new(config)?)) + Ok(Arc::new(RuntimeEnv::try_new(config)?)) }); let logical = opt diff --git a/ballista/executor/src/standalone.rs b/ballista/executor/src/standalone.rs index 28efe70fa..40f1088ec 100644 --- a/ballista/executor/src/standalone.rs +++ b/ballista/executor/src/standalone.rs @@ -170,7 +170,7 @@ pub async fn new_standalone_executor< let config = with_object_store_registry( RuntimeConfig::new().with_temp_file_path(wd.clone()), ); - Ok(Arc::new(RuntimeEnv::new(config)?)) + Ok(Arc::new(RuntimeEnv::try_new(config)?)) }); let executor = Arc::new(Executor::new_basic(