Fix pyspark tests.

vanzin · Feb 25, 2015 · 7e66c18 · 7e66c18
1 parent 0031a8e
commit 7e66c18
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bin/pyspark b/bin/pyspark
@@ -86,6 +86,7 @@ export PYTHONSTARTUP="$SPARK_HOME/python/pyspark/shell.py"
 if [[ -n "$SPARK_TESTING" ]]; then
   unset YARN_CONF_DIR
   unset HADOOP_CONF_DIR
+  export PYSPARK_SUBMIT_ARGS=pyspark-shell
   if [[ -n "$PYSPARK_DOC_TEST" ]]; then
     exec "$PYSPARK_DRIVER_PYTHON" -m doctest $1
   else