kubeflow / trainer / 25983732200

Pull #3516

github

feat: inject PET_* envs into init containers via envInjection config

- Add EnvInjection and EnvInjectionTarget types to TorchMLPolicySource
- Add FindContainerByPodSetName helper in runtime package
- Update torch plugin to inject PET_* envs into additional containers
- Update jobset plugin to sync init containers
- Add comprehensive unit tests for envInjection scenarios
- Fix torchtune detection for env injection logic

Signed-off-by: Peter Pan <Peter.Pan@daocloud.io>

41 of 72 new or added lines in 3 files covered. (56.94%)

2207 of 3566 relevant lines covered (61.89%)

0.72 hits per line