Update the code according

apache · Apr 10, 2015 · f889657 · f889657
1 parent 8a2f3e2
commit f889657
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/run-tests b/python/run-tests
@@ -113,7 +113,7 @@ function run_streaming_tests() {
       KAFKA_ASSEMBLY_JAR="$f"
     done
 
-    export EXTRA_PYSPARK_SUBMIT_ARGS="--jars ${KAFKA_ASSEMBLY_JAR}"
+    export PYSPARK_SUBMIT_ARGS="--jars ${KAFKA_ASSEMBLY_JAR} pyspark-shell"
     run_test "pyspark/streaming/util.py"
     run_test "pyspark/streaming/tests.py"
 }