kubeflow / trainer / 25990460107

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 5 comprehensive unit tests for envInjection scenarios
- Add Python API models for envInjection types

38 of 69 new or added lines in 3 files covered. (55.07%)

2201 of 3563 relevant lines covered (61.77%)

0.72 hits per line