fix imports after un-rebased PR (#1106)

apache · Oct 31, 2024 · 626e9fa · 626e9fa
1 parent 19d829f
commit 626e9fa
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -16,7 +16,7 @@
 // under the License.
 
 use ballista::{
-    extension::BallistaSessionConfigExt,
+    extension::SessionConfigExt,
     prelude::{
         Result, SessionContextExt, BALLISTA_DEFAULT_SHUFFLE_PARTITIONS,
         BALLISTA_STANDALONE_PARALLELISM,