Skip to content

Commit 7ba6b96

Browse files
committed
[SPARK-28770][CORE][TEST] Simplify the filter.
Suggested by Jungtaek Lim.
1 parent 0b0adb0 commit 7ba6b96

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

core/src/test/scala/org/apache/spark/scheduler/ReplayListenerSuite.scala

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -230,11 +230,9 @@ class ReplayListenerSuite extends SparkFunSuite with BeforeAndAfter with LocalSp
230230
// (at heartbeat intervals), and when we do, it is only in the original events, never
231231
// in the replay, since executor metrics updates are not replayed.
232232
// For this reason, exclude stage executor metrics for the driver.
233-
val filteredEvents = originalEvents.filter { e =>
234-
if (e.isInstanceOf[SparkListenerStageExecutorMetrics] &&
235-
e.asInstanceOf[SparkListenerStageExecutorMetrics].execId == "driver") {
236-
false
237-
} else true
233+
val filteredEvents = originalEvents.filter {
234+
case e: SparkListenerStageExecutorMetrics if e.execId == "driver" => false
235+
case _ => true
238236
}
239237
filteredEvents.zip(replayedEvents).foreach { case (e1, e2) =>
240238
JsonProtocolSuite.assertEquals(e1, e1)

0 commit comments

Comments
 (0)