Skip to content

vllm.v1.executor.ray_env_utils

Functions:

get_driver_env_vars(worker_specific_vars)

Return driver env vars to propagate to Ray workers.

Returns everything from os.environ except worker_specific_vars and user-configured exclusions (RAY_NON_CARRY_OVER_ENV_VARS).

Source code in vllm/v1/executor/ray_env_utils.py 
 8
 9
10
11
12
13
14
15
16
17
18
def get_driver_env_vars(
    worker_specific_vars: set[str],
) -> dict[str, str]:
    """Return driver env vars to propagate to Ray workers.

    Returns everything from ``os.environ`` except ``worker_specific_vars``
    and user-configured exclusions (``RAY_NON_CARRY_OVER_ENV_VARS``).
    """
    exclude_vars = worker_specific_vars | RAY_NON_CARRY_OVER_ENV_VARS

    return {key: value for key, value in os.environ.items() if key not in exclude_vars}