replace deprecations

apache · Nov 8, 2024 · f3769a6 · f3769a6
1 parent 3229367
commit f3769a6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git 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<T: 'static + AsLogicalPlan>(
         .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
@@ -203,7 +203,7 @@ pub async fn start_executor_process(opt: Arc<ExecutorProcessConfig>) -> 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
@@ -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(