pub expose datafusion in ballista client

this will simplify basic usage where we wont need to include datafusion as dependency
milenkovicm · Jan 22, 2025 · 312417f · 312417f
1 parent 8539c49
commit 312417f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/ballista/client/src/lib.rs b/ballista/client/src/lib.rs
@@ -19,3 +19,4 @@
 
 pub mod extension;
 pub mod prelude;
+pub use datafusion;
diff --git a/examples/examples/remote-sql.rs b/examples/examples/remote-sql.rs
@@ -15,13 +15,13 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use ballista::prelude::*;
-use ballista_examples::test_util;
-use datafusion::{
+use ballista::datafusion::{
     common::Result,
     execution::SessionStateBuilder,
     prelude::{CsvReadOptions, SessionConfig, SessionContext},
 };
+use ballista::prelude::*;
+use ballista_examples::test_util;
 
 /// This example demonstrates executing a simple query against an Arrow data source (CSV) and
 /// fetching results, using SQL

diff --git a/examples/examples/standalone-sql.rs b/examples/examples/standalone-sql.rs
@@ -15,13 +15,13 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use ballista::prelude::{SessionConfigExt, SessionContextExt};
-use ballista_examples::test_util;
-use datafusion::{
+use ballista::datafusion::{
     common::Result,
     execution::{options::ParquetReadOptions, SessionStateBuilder},
     prelude::{SessionConfig, SessionContext},
 };
+use ballista::prelude::{SessionConfigExt, SessionContextExt};
+use ballista_examples::test_util;
 
 #[tokio::main]
 async fn main() -> Result<()> {
Original file line number	Diff line number	Diff line change
Expand Up		@@ -19,3 +19,4 @@

		pub mod extension;
		pub mod prelude;
		pub use datafusion;