Fix shuffle writer test (#998)

* Fix test data in shuffle writer to produce 2 partitions * Remove TODO * Clippy
apache · Apr 13, 2024 · 321af95 · 321af95
1 parent 1be0a46
commit 321af95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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,
             }];