sarutak commented on a change in pull request #28839: URL: https://github.com/apache/spark/pull/28839#discussion_r440966224
########## File path: core/src/test/scala/org/apache/spark/scheduler/BarrierTaskContextSuite.scala ########## @@ -275,7 +275,8 @@ class BarrierTaskContextSuite extends SparkFunSuite with LocalSparkContext with } test("SPARK-31485: barrier stage should fail if only partial tasks are launched") { - initLocalClusterSparkContext(2) + val conf = new SparkConf().set(LOCALITY_WAIT_PROCESS.key, Int.MaxValue + "s") Review comment: This testcase requires both two tasks should be assigned to the same preferred location so `Int.MaxValue` means the rest of task which is not assigned the preferred location first waits for the preferred location available. If it's concerned to hung the test, we can choose a small value but long enough to assign to the preferred location. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org