Set max retry to 1 only for interactive sessions.

cloudera · Jul 12, 2016 · 39a5162 · 39a5162
1 parent 9d77ee0
commit 39a5162
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/server/src/main/scala/com/cloudera/livy/server/interactive/InteractiveSession.scala b/server/src/main/scala/com/cloudera/livy/server/interactive/InteractiveSession.scala
@@ -69,6 +69,9 @@ object InteractiveSession {
     }
 
     val create = { (s: InteractiveSession) =>
+      if (livyConf.isSparkMasterYarn) {
+        builder.conf("spark.yarn.maxAppAttempts", "1")
+      }
       SparkApp.create(s.uuid, builder, None, List(kindString), livyConf, Option(s))
     }
 

diff --git a/server/src/main/scala/com/cloudera/livy/utils/SparkApp.scala b/server/src/main/scala/com/cloudera/livy/utils/SparkApp.scala
@@ -70,7 +70,6 @@ object SparkApp extends Logging {
     if (livyConf.isSparkMasterYarn) {
       val appTag = uuidToAppTag(uuid)
       builder.conf("spark.yarn.tags", appTag)
-      builder.conf("spark.yarn.maxAppAttempts", "1")
 
       listener.foreach(_.startingApp())
       val process: Option[LineBufferedProcess] = {