fix: Fix the driver concurrency setting in test (#11805)

Summary: Pull Request resolved: #11805 The tests use a non-existing query config Reviewed By: kagamiori Differential Revision: D66994115 fbshipit-source-id: 4b7a2bd56f5bf2c6fa8b2236eb1478a0e058b3fa
facebookincubator · Dec 10, 2024 · 4f863c7 · 4f863c7
1 parent 386fed7
commit 4f863c7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/velox/exec/tests/AggregationTest.cpp b/velox/exec/tests/AggregationTest.cpp
@@ -941,7 +941,7 @@ TEST_F(AggregationTest, partialDistinctWithAbandon) {
   auto task = AssertQueryBuilder(duckDbQueryRunner_)
                   .config(QueryConfig::kAbandonPartialAggregationMinRows, 100)
                   .config(QueryConfig::kAbandonPartialAggregationMinPct, 50)
-                  .config("max_drivers_per_task", 1)
+                  .maxDrivers(1)
                   .plan(PlanBuilder()
                             .values(vectors)
                             .partialAggregation({"c0"}, {})
@@ -953,7 +953,7 @@ TEST_F(AggregationTest, partialDistinctWithAbandon) {
   task = AssertQueryBuilder(duckDbQueryRunner_)
              .config(QueryConfig::kAbandonPartialAggregationMinRows, 100)
              .config(QueryConfig::kAbandonPartialAggregationMinPct, 50)
-             .config("max_drivers_per_task", 1)
+             .maxDrivers(1)
              .plan(PlanBuilder()
                        .values(vectors)
                        .partialAggregation({"c0"}, {"sum(c0)"})
@@ -990,7 +990,7 @@ TEST_F(AggregationTest, distinctWithGroupingKeysReordered) {
                   .config(QueryConfig::kSpillEnabled, true)
                   .config(QueryConfig::kAggregationSpillEnabled, true)
                   .config(QueryConfig::kSpillPrefixSortEnabled, true)
-                  .config("max_drivers_per_task", 1)
+                  .maxDrivers(1)
                   .plan(PlanBuilder()
                             .values(vectors)
                             .singleAggregation({"c2", "c0"}, {})