Set spark.executor.uri from environment variable (needed by Mesos)

witgo · Apr 3, 2014 · da0c3e4 · da0c3e4
1 parent 47ebea5
commit da0c3e4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/pyspark/shell.py b/python/pyspark/shell.py
@@ -29,6 +29,9 @@
 # this is the equivalent of ADD_JARS
 add_files = os.environ.get("ADD_FILES").split(',') if os.environ.get("ADD_FILES") != None else None
 
+if os.environ.get("SPARK_EXECUTOR_URI"):
+    SparkContext.setSystemProperty("spark.executor.uri", os.environ["SPARK_EXECUTOR_URI"])
+
 sc = SparkContext(os.environ.get("MASTER", "local"), "PySparkShell", pyFiles=add_files)
 
 print """Welcome to