From b0ecf1ee41f2280a885a8005363e0d0b197e4d3b Mon Sep 17 00:00:00 2001 From: Matei Zaharia Date: Mon, 27 Jun 2011 22:50:43 -0700 Subject: Don't pass a null context when running tasks locally --- core/src/main/scala/spark/DAGScheduler.scala | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'core') diff --git a/core/src/main/scala/spark/DAGScheduler.scala b/core/src/main/scala/spark/DAGScheduler.scala index 99a69203af..ecdce037e9 100644 --- a/core/src/main/scala/spark/DAGScheduler.scala +++ b/core/src/main/scala/spark/DAGScheduler.scala @@ -171,7 +171,8 @@ private trait DAGScheduler extends Scheduler with Logging { if (finalStage.parents.size == 0 && numOutputParts == 1) { logInfo("Computing the requested partition locally") val split = finalRdd.splits(outputParts(0)) - return Array(func(null, finalRdd.iterator(split))) + val taskContext = new TaskContext(finalStage.id, outputParts(0), 0) + return Array(func(taskContext, finalRdd.iterator(split))) } def submitStage(stage: Stage) { -- cgit v1.2.3