Fix env vars for running Ray distributed backend on GKE (#15166)

Signed-off-by: Richard Liu <ricliu@google.com>
This commit is contained in:
Richard Liu
2025-03-20 07:59:33 -07:00
committed by GitHub
parent 69ae2380c6
commit a8f12a63fd
3 changed files with 8 additions and 0 deletions

View File

@@ -340,6 +340,8 @@ class RayDistributedExecutor(DistributedExecutorBase):
and v not in self.non_carry_over_env_vars
]
env_vars_to_copy.extend(current_platform.additional_env_vars)
# Copy existing env vars to each worker's args
for args in all_args_to_update_environment_variables:
# TODO: refactor platform-specific env vars