We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f87802f commit 5ef6846Copy full SHA for 5ef6846
jvm-packages/xgboost4j-spark/src/main/scala/org/apache/spark/SparkParallelismTracker.scala
@@ -76,11 +76,12 @@ class SparkParallelismTracker(
76
}
77
78
private[this] def safeExecute[T](body: => T): T = {
79
- sc.listenerBus.listeners.add(0, new TaskFailedListener)
+ val listener = new TaskFailedListener;
80
+ sc.addSparkListener(listener)
81
try {
82
body
83
} finally {
- sc.listenerBus.listeners.remove(0)
84
+ sc.listenerBus.removeListener(listener)
85
86
87
0 commit comments