From 929affe389155af6e792154f3e44f9019e5dd320 Mon Sep 17 00:00:00 2001
From: Xiaoxuan Meng <xiaoxmeng@meta.com>
Date: Sat, 7 Dec 2024 21:38:07 -0800
Subject: [PATCH] fix: Fix flaky caleWriterLocalPartitionTest.unpartitionBasic
 (#11788)

Summary:
Pull Request resolved: https://github.com/facebookincubator/velox/pull/11788

Fix flaky caleWriterLocalPartitionTest.unpartitionBasic witb more input data

Reviewed By: tanjialiang

Differential Revision: D66918241

fbshipit-source-id: f000c456949a4aa66a0864194ea07360f95091a9
---
 velox/exec/tests/ScaleWriterLocalPartitionTest.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp b/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
index b0fbe45f470e..5ea6a5715266 100644
--- a/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
+++ b/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
@@ -552,7 +552,7 @@ class ScaleWriterLocalPartitionTest : public HiveConnectorTestBase {
 };
 
 TEST_F(ScaleWriterLocalPartitionTest, unpartitionBasic) {
-  const std::vector<RowVectorPtr> inputVectors = makeVectors(32, 1024);
+  const std::vector<RowVectorPtr> inputVectors = makeVectors(128, 1024);
   const uint64_t queryCapacity = 256 << 20;
   const uint32_t maxDrivers = 32;
   const uint32_t maxExchanegBufferSize = 2 << 20;