From 626e9fa4a8bd86e7493c758a5888a57c27b3dcb2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marko=20Milenkovi=C4=87?= Date: Thu, 31 Oct 2024 22:02:52 +0000 Subject: [PATCH] fix imports after un-rebased PR (#1106) --- ballista-cli/src/main.rs | 2 +- ballista/client/src/prelude.rs | 2 +- benchmarks/src/bin/tpch.rs | 2 +- examples/examples/remote-dataframe.rs | 2 +- examples/examples/remote-sql.rs | 2 +- examples/examples/standalone-sql.rs | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/ballista-cli/src/main.rs b/ballista-cli/src/main.rs index ec0c5bbca..1e94a7339 100644 --- a/ballista-cli/src/main.rs +++ b/ballista-cli/src/main.rs @@ -19,7 +19,7 @@ use std::env; use std::path::Path; use ballista::{ - extension::BallistaSessionConfigExt, + extension::SessionConfigExt, prelude::{ Result, SessionContextExt, BALLISTA_DEFAULT_BATCH_SIZE, BALLISTA_STANDALONE_PARALLELISM, BALLISTA_WITH_INFORMATION_SCHEMA, diff --git a/ballista/client/src/prelude.rs b/ballista/client/src/prelude.rs index d06a65301..bdac712bb 100644 --- a/ballista/client/src/prelude.rs +++ b/ballista/client/src/prelude.rs @@ -30,5 +30,5 @@ pub use ballista_core::{ #[allow(deprecated)] // TO BE REMOVED pub use crate::context::BallistaContext; -pub use crate::extension::SessionContextExt; +pub use crate::extension::{SessionConfigExt, SessionContextExt}; pub use futures::StreamExt; diff --git a/benchmarks/src/bin/tpch.rs b/benchmarks/src/bin/tpch.rs index c38062776..068b61c2d 100644 --- a/benchmarks/src/bin/tpch.rs +++ b/benchmarks/src/bin/tpch.rs @@ -18,7 +18,7 @@ //! Benchmark derived from TPC-H. This is not an official TPC-H benchmark. use arrow_schema::SchemaBuilder; -use ballista::extension::BallistaSessionConfigExt; +use ballista::extension::SessionConfigExt; use ballista::prelude::{ SessionContextExt, BALLISTA_COLLECT_STATISTICS, BALLISTA_DEFAULT_BATCH_SIZE, BALLISTA_DEFAULT_SHUFFLE_PARTITIONS, BALLISTA_JOB_NAME, diff --git a/examples/examples/remote-dataframe.rs b/examples/examples/remote-dataframe.rs index ae6d65319..db2316e5c 100644 --- a/examples/examples/remote-dataframe.rs +++ b/examples/examples/remote-dataframe.rs @@ -15,7 +15,7 @@ // specific language governing permissions and limitations // under the License. -use ballista::{extension::BallistaSessionConfigExt, prelude::*}; +use ballista::{extension::SessionConfigExt, prelude::*}; use ballista_examples::test_util; use datafusion::{ execution::SessionStateBuilder, diff --git a/examples/examples/remote-sql.rs b/examples/examples/remote-sql.rs index 26ed11bc7..756791ec4 100644 --- a/examples/examples/remote-sql.rs +++ b/examples/examples/remote-sql.rs @@ -15,7 +15,7 @@ // specific language governing permissions and limitations // under the License. -use ballista::{extension::BallistaSessionConfigExt, prelude::*}; +use ballista::{extension::SessionConfigExt, prelude::*}; use ballista_examples::test_util; use datafusion::{ execution::SessionStateBuilder, diff --git a/examples/examples/standalone-sql.rs b/examples/examples/standalone-sql.rs index d1e637e7c..5b9632532 100644 --- a/examples/examples/standalone-sql.rs +++ b/examples/examples/standalone-sql.rs @@ -16,7 +16,7 @@ // under the License. use ballista::{ - extension::BallistaSessionConfigExt, + extension::SessionConfigExt, prelude::{ Result, SessionContextExt, BALLISTA_DEFAULT_SHUFFLE_PARTITIONS, BALLISTA_STANDALONE_PARALLELISM,