From 33d9fc9a8ba8d3499bd1251c7bcf4d862243320e Mon Sep 17 00:00:00 2001 From: architkulkarni Date: Fri, 21 May 2021 09:02:32 -0700 Subject: [PATCH] [Core] minor refactor for usability of runtime_env (#15965) * minor refactor for usability of runtime_env * remove job_config changes --- python/ray/_private/runtime_env.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/ray/_private/runtime_env.py b/python/ray/_private/runtime_env.py index a2bab0048..6e1af892d 100644 --- a/python/ray/_private/runtime_env.py +++ b/python/ray/_private/runtime_env.py @@ -117,6 +117,13 @@ class RuntimeEnvDict: self._dict["working_dir"] = runtime_env_json["working_dir"] else: self._dict["working_dir"] = None + + if "uris" in runtime_env_json: + self._dict["uris"] = runtime_env_json["uris"] + + if "_ray_release" in runtime_env_json: + self._dict["_ray_release"] = runtime_env_json["_ray_release"] + # TODO(ekl) we should have better schema validation here. # TODO(ekl) support py_modules # TODO(architkulkarni) support env_vars, docker