[Backport 1.9.latest] Fix microbatch model PartialSuccess status to result in non-zero exit code

[Backport 1.9.latest] Fix microbatch model PartialSuccess status to result in non-zero exit code #5245