switch back

apache · Sep 29, 2014 · 32f4dfa · 32f4dfa
1 parent f0b6519
commit 32f4dfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yarn/common/src/main/scala/org/apache/spark/deploy/yarn/ApplicationMaster.scala b/yarn/common/src/main/scala/org/apache/spark/deploy/yarn/ApplicationMaster.scala
@@ -113,8 +113,8 @@ private[spark] class ApplicationMaster(args: ApplicationMasterArguments,
           if (!unregistered) {
             // we only want to unregister if we don't want the RM to retry
             if (finalStatus == FinalApplicationStatus.SUCCEEDED || isLastAttempt) {
-              cleanupStagingDir()
               unregister(finalStatus, finalMsg)
+              cleanupStagingDir()
             }
           }
         }