change order of cleanup staging dir

apache · Sep 29, 2014 · f0b6519 · f0b6519
1 parent d3cc800
commit f0b6519
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) {
-              unregister(finalStatus, finalMsg)
               cleanupStagingDir()
+              unregister(finalStatus, finalMsg)
             }
           }
         }