Skip to content
Snippets Groups Projects
Commit 7e8d2e8a authored by Matei Zaharia's avatar Matei Zaharia
Browse files

Fix Python code after change of getOrElse

parent 0f606073
No related branches found
No related tags found
No related merge requests found
......@@ -134,7 +134,12 @@ class SparkConf(object):
def get(self, key, defaultValue=None):
"""Get the configured value for some key, or return a default otherwise."""
return self._jconf.get(key, defaultValue)
if defaultValue == None: # Py4J doesn't call the right get() if we pass None
if not self._jconf.contains(key):
return None
return self._jconf.get(key)
else:
return self._jconf.get(key, defaultValue)
def getAll(self):
"""Get all values as a list of key-value pairs."""
......
......@@ -92,11 +92,13 @@ class SparkContext(object):
self.serializer = BatchedSerializer(self._unbatched_serializer,
batchSize)
# Set parameters passed directly to us on the conf; these operations will be
# no-ops if the parameters were None
self._conf.setMaster(master)
self._conf.setAppName(appName)
self._conf.setSparkHome(sparkHome)
# Set any parameters passed directly to us on the conf
if master:
self._conf.setMaster(master)
if appName:
self._conf.setAppName(appName)
if sparkHome:
self._conf.setSparkHome(sparkHome)
if environment:
for key, value in environment.iteritems():
self._conf.setExecutorEnv(key, value)
......@@ -111,7 +113,7 @@ class SparkContext(object):
# the classpath or an external config file
self.master = self._conf.get("spark.master")
self.appName = self._conf.get("spark.app.name")
self.sparkHome = self._conf.getOrElse("spark.home", None)
self.sparkHome = self._conf.get("spark.home", None)
for (k, v) in self._conf.getAll():
if k.startswith("spark.executorEnv."):
varName = k[len("spark.executorEnv."):]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment