kubeflow / trainer / 17403349307

Pull #2653

github

fix(runtime): prevent launcher config override when runLauncherAsNode is true

Previously, when runLauncherAsNode was set to true, the launcher
container would receive the full trainer configuration including
image, command, args, and environment variables, which could
unintentionally override the launcher's original configuration.

This change ensures that launcher containers only receive resource
allocations (CPU, memory, GPU) when runLauncherAsNode is true, while
preserving their original image, command, and args. Node containers
continue to receive the full trainer configuration as expected.

The fix avoids override issues and improves the separation of
concerns between launcher and node container configurations in
MPI-based training jobs.

- Refactor Trainer method to apply full config only to Node containers
- Apply resources separately to both Node and Launcher containers
- Add comprehensive test case to verify launcher behavior
- Update existing test expectations to match correct behavior

Signed-off-by: Krishnaswamy Subramanian <subramk@thoughtworks.com>

35 of 38 new or added lines in 1 file covered. (92.11%)

1081 of 1975 relevant lines covered (54.73%)

0.65 hits per line