diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml index 1cee5139e3fbd..37d8ada344bdf 100644 --- a/.github/workflows/rust.yml +++ b/.github/workflows/rust.yml @@ -75,6 +75,12 @@ jobs: env: CARGO_HOME: "/github/home/.cargo" CARGO_TARGET_DIR: "/github/home/target" + - name: Check DataFusion builds with all features + run: | + cargo check --workspace --benches --features avro,jit,scheduler -p datafusion + env: + CARGO_HOME: "/github/home/.cargo" + CARGO_TARGET_DIR: "/github/home/target" # test the crate linux-test: @@ -115,7 +121,7 @@ jobs: run: | export ARROW_TEST_DATA=$(pwd)/testing/data export PARQUET_TEST_DATA=$(pwd)/parquet-testing/data - cargo test --features avro + cargo test --features avro,jit,scheduler # test datafusion-sql examples cargo run --example sql # test datafusion examples diff --git a/datafusion/core/benches/parquet_query_sql.rs b/datafusion/core/benches/parquet_query_sql.rs index de4dcf66a9d44..87f14114f7c81 100644 --- a/datafusion/core/benches/parquet_query_sql.rs +++ b/datafusion/core/benches/parquet_query_sql.rs @@ -260,8 +260,10 @@ fn criterion_benchmark(c: &mut Criterion) { local_rt.block_on(async { let query = context.sql(&query).await.unwrap(); let plan = query.create_physical_plan().await.unwrap(); - let mut stream = + let results = scheduler.schedule(plan, context.task_ctx()).unwrap(); + + let mut stream = results.stream(); while stream.next().await.transpose().unwrap().is_some() {} }); });