fix: Fix the driver concurrency setting in test

Summary: The tests use a non-existing query config Differential Revision: D66994115
xiaoxmeng · Dec 10, 2024 · 8f6ef57 · 8f6ef57
1 parent b9cce6d
commit 8f6ef57
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"}, {})