12345678910111213141516171819202122232425262728293031 |
- import os
- from typing import Dict
- from ray_release.bazel import bazel_runfile
- from ray_release.exception import ReleaseTestConfigError
- DEFAULT_ENVIRONMENT = "aws"
- def load_environment(environment_name: str) -> Dict[str, str]:
- file_base = f"{environment_name}.env"
- env_file = bazel_runfile("release/ray_release/environments", file_base)
- if not env_file or not os.path.isfile(env_file):
- raise ReleaseTestConfigError(
- f"Unknown environment with name: {environment_name}"
- )
- env = {}
- with open(env_file, "r") as f:
- for line in f.readlines():
- if not line:
- continue
- key, val = line.strip().split("=", maxsplit=1)
- env[key] = val.strip('"')
- return env
- def populate_os_env(env: Dict[str, str]) -> None:
- for k, v in env.items():
- os.environ[k] = v
|