[Core] minor refactor for usability of runtime_env (#15965)

* minor refactor for usability of runtime_env

* remove job_config changes
This commit is contained in:
architkulkarni 2021-05-21 09:02:32 -07:00 committed by Dmitri Gekhtman
parent 44d355aeea
commit 33d9fc9a8b

View file

@ -117,6 +117,13 @@ class RuntimeEnvDict:
self._dict["working_dir"] = runtime_env_json["working_dir"] self._dict["working_dir"] = runtime_env_json["working_dir"]
else: else:
self._dict["working_dir"] = None 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) we should have better schema validation here.
# TODO(ekl) support py_modules # TODO(ekl) support py_modules
# TODO(architkulkarni) support env_vars, docker # TODO(architkulkarni) support env_vars, docker