Catch Throwable instead of Exception in LocalScheduler and Executor. Fixes #57.

2 files changed, 4 insertions, 4 deletions

diff --git a/core/src/main/scala/spark/Executor.scala b/core/src/main/scala/spark/Executor.scala
index 7764acad1a..54e169a1a1 100644
--- a/core/src/main/scala/spark/Executor.scala
+++ b/core/src/main/scala/spark/Executor.scala
@@ -69,9 +69,9 @@ class Executor extends mesos.Executor with Logging {
           d.sendStatusUpdate(new TaskStatus(
             taskId, TaskState.TASK_FAILED, Utils.serialize(reason)))
         }
-        case e: Exception => {
+        case t: Throwable => {
           // TODO: Handle errors in tasks less dramatically
-          logError("Exception in task ID " + taskId, e)
+          logError("Exception in task ID " + taskId, t)
           System.exit(1)
         }
       }
diff --git a/core/src/main/scala/spark/LocalScheduler.scala b/core/src/main/scala/spark/LocalScheduler.scala
index 88da0c1af1..7d3ab661aa 100644
--- a/core/src/main/scala/spark/LocalScheduler.scala
+++ b/core/src/main/scala/spark/LocalScheduler.scala
@@ -36,9 +36,9 @@ private class LocalScheduler(threads: Int) extends DAGScheduler with Logging {
             logInfo("Finished task " + i)
             taskEnded(tasks(i), Success, result, accumUpdates)
           } catch {
-            case e: Exception => {
+            case t: Throwable => {
               // TODO: Do something nicer here
-              logError("Exception in task " + i, e)
+              logError("Exception in task " + i, t)
               System.exit(1)
               null
             }