Refine comment

Author: caneGuy

core/src/main/scala/org/apache/spark/scheduler/DAGScheduler.scala

@@ -1045,14 +1045,9 @@ class DAGScheduler(
 
         // Abort execution
         return
-      case NonFatal(e) =>
+      case e: Throwable =>
         abortStage(stage, s"Task serialization failed: $e\n${Utils.exceptionString(e)}", Some(e))
         runningStages -= stage
-        return
-
-      case e: NoClassDefFoundError =>
-        abortStage(stage, s"Task serialization failed:  $e\n${Utils.exceptionString(e)}", Some(e))
-        runningStages -= stage
 
         // Abort execution
         return