Fix flaky test TaskTest.inconsistentExecutionMode

tanjialiang · Jun 3, 2024 · 05ccb53 · 05ccb53
1 parent ab06a77
commit 05ccb53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/velox/exec/tests/TaskTest.cpp b/velox/exec/tests/TaskTest.cpp
@@ -1278,6 +1278,7 @@ DEBUG_ONLY_TEST_F(TaskTest, inconsistentExecutionMode) {
     while (cursor->hasNext()) {
       cursor->moveNext();
     }
+    waitForTaskCompletion(task);
   }
 
   {