chore: import upstream snapshot with attribution

release/ray_release/env.py

@@ -0,0 +1,31 @@
+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