From c7092d5a1955796176ec411f3cc2bd85f5f61b93 Mon Sep 17 00:00:00 2001 From: Ian Rodney Date: Fri, 21 May 2021 14:57:44 -0700 Subject: [PATCH] Revert "Revert "[Core] minor refactor for usability of runtime_env (#15965)" (#15985)" (#15987) --- python/build-wheel-manylinux2014.sh | 5 ++++- python/ray/_private/runtime_env.py | 7 +++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/python/build-wheel-manylinux2014.sh b/python/build-wheel-manylinux2014.sh index 490ec256e..54215fa8d 100755 --- a/python/build-wheel-manylinux2014.sh +++ b/python/build-wheel-manylinux2014.sh @@ -95,7 +95,10 @@ done # hack, we should use auditwheel instead. for path in .whl/*.whl; do if [ -f "${path}" ]; then - mv "${path}" "${path//linux/manylinux2014}" + out="${path//-linux/-manylinux2014}" + if [ "$out" != "$path" ]; then + mv "${path}" "${out}" + fi fi done 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