get rid of SparkIllegalStateException

Author: squito

core/src/main/scala/org/apache/spark/SparkException.scala

@@ -30,13 +30,3 @@ class SparkException(message: String, cause: Throwable)
  */
 private[spark] class SparkDriverExecutionException(cause: Throwable)
   extends SparkException("Execution error", cause)
-
-/**
- * Exception indicating an error internal to Spark -- it is in an inconsistent state, not due
- * to any error by the user
- */
-class SparkIllegalStateException(message: String, cause: Throwable)
-  extends SparkException(message, cause) {
-
-  def this(message: String) = this(message, null)
-}

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

@@ -171,7 +171,7 @@ private[spark] class TaskSchedulerImpl(
         ts.taskSet != taskSet && !ts.isZombie
       }
       if (conflictingTaskSet) {
-        throw new SparkIllegalStateException(s"more than one active taskSet for stage $stage:" +
+        throw new IllegalStateException(s"more than one active taskSet for stage $stage:" +
           s" ${stageTaskSets.toSeq.map{_._2.taskSet.id}.mkString(",")}")
       }
       schedulableBuilder.addTaskSetManager(manager, manager.taskSet.properties)

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

@@ -141,13 +141,13 @@ class TaskSchedulerImplSuite extends SparkFunSuite with LocalSparkContext with L
     val attempt1 = new TaskSet(Array(new FakeTask(0)), 0, 0, 0, null)
     val attempt2 = new TaskSet(Array(new FakeTask(0)), 0, 1, 0, null)
     taskScheduler.submitTasks(attempt1)
-    intercept[SparkIllegalStateException] { taskScheduler.submitTasks(attempt2) }
+    intercept[IllegalStateException] { taskScheduler.submitTasks(attempt2) }
 
     // OK to submit multiple if previous attempts are all zombie
     taskScheduler.activeTaskSets(attempt1.id).isZombie = true
     taskScheduler.submitTasks(attempt2)
     val attempt3 = new TaskSet(Array(new FakeTask(0)), 0, 2, 0, null)
-    intercept[SparkIllegalStateException] { taskScheduler.submitTasks(attempt3) }
+    intercept[IllegalStateException] { taskScheduler.submitTasks(attempt3) }
     taskScheduler.activeTaskSets(attempt2.id).isZombie = true
     taskScheduler.submitTasks(attempt3)
   }