Browse Source

[fix] mlflow project success status (#10897)

close: #10902
3.1.0-release
JieguangZhou 2 years ago committed by GitHub
parent
commit
02b2de2a1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      dolphinscheduler-task-plugin/dolphinscheduler-task-mlflow/src/main/java/org/apache/dolphinscheduler/plugin/task/mlflow/MlflowTask.java

4
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 // construct process
String command = buildCommand(); String command = buildCommand();
TaskResponse commandExecuteResult = shellCommandExecutor.run(command); TaskResponse commandExecuteResult = shellCommandExecutor.run(command);
int exitCode = exitStatusCode; int exitCode;
if (mlflowParameters.getIsDeployDocker()){ if (mlflowParameters.getIsDeployDocker()){
exitCode = checkDockerHealth(); exitCode = checkDockerHealth();
}else { }else {
exitCode = getExitStatusCode(); exitCode = commandExecuteResult.getExitStatusCode();
} }
setExitStatusCode(exitCode); setExitStatusCode(exitCode);
setAppIds(commandExecuteResult.getAppIds()); setAppIds(commandExecuteResult.getAppIds());

Loading…
Cancel
Save