applying review comments

apache · Jan 19, 2018 · cf6e0c9 · cf6e0c9
1 parent 41dd7bb
commit cf6e0c9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/core/src/main/scala/org/apache/spark/status/AppStatusListener.scala b/core/src/main/scala/org/apache/spark/status/AppStatusListener.scala
@@ -212,7 +212,7 @@ private[spark] class AppStatusListener(
   }
 
   override def onExecutorBlacklistedForStage(
-    event: SparkListenerExecutorBlacklistedForStage): Unit = {
+      event: SparkListenerExecutorBlacklistedForStage): Unit = {
     updateBlackListStatusForStage(event.executorId, event.stageId, event.stageAttemptId)
   }
 

diff --git a/core/src/test/scala/org/apache/spark/scheduler/TaskSetBlacklistSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/TaskSetBlacklistSuite.scala
@@ -87,8 +87,6 @@ class TaskSetBlacklistSuite extends SparkFunSuite with BeforeAndAfterEach with M
     assert(taskSetBlacklist.isExecutorBlacklistedForTaskSet("exec1"))
 
     assert(!taskSetBlacklist.isExecutorBlacklistedForTaskSet("exec2"))
-    verify(listenerBusMock, never()).post(
-      SparkListenerNodeBlacklistedForStage(0, "hostA", 2, 0, attemptId))
 
     assert(!taskSetBlacklist.isNodeBlacklistedForTaskSet("hostA"))
     verify(listenerBusMock, never())