Added exception handling instead of just exiting in LocalScheduler for tasks that throw exceptions

1 files changed, 2 insertions, 1 deletions

diff --git a/core/src/main/scala/spark/LocalScheduler.scala b/core/src/main/scala/spark/LocalScheduler.scala
index c86b967aa5..c675eb5462 100644
--- a/core/src/main/scala/spark/LocalScheduler.scala
+++ b/core/src/main/scala/spark/LocalScheduler.scala
@@ -58,7 +58,8 @@ private class LocalScheduler(threads: Int, maxFailures: Int) extends DAGSchedule
               submitTask(task, idInJob)
             } else {
               // TODO: Do something nicer here to return all the way to the user
-              System.exit(1)
+              //System.exit(1)
+              taskEnded(task, new ExceptionFailure(t), null, null)
             }
           }
         }