Unsetting the TaskContext which was set with empty TaskContext

apache · Jun 24, 2017 · 15b77df · 15b77df
1 parent 371933d
commit 15b77df
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala b/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala
@@ -1108,5 +1108,6 @@ class UtilsSuite extends SparkFunSuite with ResetSystemProperties with Logging {
     assert(isErrorOccurred)
     // if the try, catch and finally blocks don't throw exceptions
     Utils.tryWithSafeFinallyAndFailureCallbacks {}(catchBlock = {}, finallyBlock = {})
+    TaskContext.unset
   }
 }