(arrow-ballista) branch main updated: Fix shuffle writer test (#998)

[agrove]

This is an automated email from the ASF dual-hosted git repository.

agrove pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/arrow-ballista.git


The following commit(s) were added to refs/heads/main by this push:
     new 321af95d Fix shuffle writer test (#998)
321af95d is described below

commit 321af95db65a522ae2e618295e1caa77cedf8c13
Author: Jeffrey Vo <jeffrey.vo.austra...@gmail.com>
AuthorDate: Sun Apr 14 07:49:27 2024 +1000

    Fix shuffle writer test (#998)
    
    * Fix test data in shuffle writer to produce 2 partitions
    
    * Remove TODO
    
    * Clippy
---
 ballista/core/src/execution_plans/shuffle_writer.rs | 2 +-
 ballista/scheduler/src/scheduler_server/grpc.rs     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ballista/core/src/execution_plans/shuffle_writer.rs 
b/ballista/core/src/execution_plans/shuffle_writer.rs
index 19d35477..b9e25ed5 100644
--- a/ballista/core/src/execution_plans/shuffle_writer.rs
+++ b/ballista/core/src/execution_plans/shuffle_writer.rs
@@ -577,7 +577,7 @@ mod tests {
         let batch = RecordBatch::try_new(
             schema.clone(),
             vec![
-                Arc::new(UInt32Array::from(vec![Some(1), Some(2)])),
+                Arc::new(UInt32Array::from(vec![Some(1), Some(3)])),
                 Arc::new(StringArray::from(vec![Some("hello"), 
Some("world")])),
             ],
         )?;
diff --git a/ballista/scheduler/src/scheduler_server/grpc.rs 
b/ballista/scheduler/src/scheduler_server/grpc.rs
index c1ff35de..2d759fb7 100644
--- a/ballista/scheduler/src/scheduler_server/grpc.rs
+++ b/ballista/scheduler/src/scheduler_server/grpc.rs
@@ -116,7 +116,7 @@ impl<T: 'static + AsLogicalPlan, U: 'static + 
AsExecutionPlan> SchedulerGrpc
                     Status::internal(msg)
                 })?;
 
-            let mut available_slots = vec![AvailableTaskSlots {
+            let mut available_slots = [AvailableTaskSlots {
                 executor_id,
                 slots: num_free_slots,
             }];