From 02b2de2a1e3198b96eb5b6a4c42411ec265e1731 Mon Sep 17 00:00:00 2001
From: JieguangZhou <jieguang_zhou@163.com>
Date: Tue, 12 Jul 2022 20:58:53 +0800
Subject: [PATCH] [fix] mlflow project success status (#10897)

close: #10902
---
 .../dolphinscheduler/plugin/task/mlflow/MlflowTask.java       | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-mlflow/src/main/java/org/apache/dolphinscheduler/plugin/task/mlflow/MlflowTask.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-mlflow/src/main/java/org/apache/dolphinscheduler/plugin/task/mlflow/MlflowTask.java
index 88da235ff0..6c87162354 100644
--- a/dolphinscheduler-task-plugin/dolphinscheduler-task-mlflow/src/main/java/org/apache/dolphinscheduler/plugin/task/mlflow/MlflowTask.java
+++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-mlflow/src/main/java/org/apache/dolphinscheduler/plugin/task/mlflow/MlflowTask.java
@@ -86,11 +86,11 @@ public class MlflowTask extends AbstractTaskExecutor {
             // construct process
             String command = buildCommand();
             TaskResponse commandExecuteResult = shellCommandExecutor.run(command);
-            int exitCode = exitStatusCode;
+            int exitCode;
             if (mlflowParameters.getIsDeployDocker()){
                 exitCode = checkDockerHealth();
             }else {
-                exitCode = getExitStatusCode();
+                exitCode = commandExecuteResult.getExitStatusCode();
             }
             setExitStatusCode(exitCode);
             setAppIds(commandExecuteResult.getAppIds());