[runtime_env] Fix failing wheel_urls release test (ray-project#31239)

This PR updates the release tests wheel_urls to make it compatible with the internal API change in PR ray-project#30970. Previously the release test was breaking with File "/home/ray/anaconda3/lib/python3.7/site-packages/ray/_private/utils.py", line 1266, in get_wheel_filename assert py_version in ray_constants.RUNTIME_ENV_CONDA_PY_VERSIONS, py_version AssertionError: 36 Signed-off-by: tmynn <[email protected]>
ju2ez · Jan 25, 2023 · 9120b06 · 9120b06
1 parent 6c7c9ae
commit 9120b06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/release/runtime_env_tests/workloads/wheel_urls.py b/release/runtime_env_tests/workloads/wheel_urls.py
@@ -21,6 +21,7 @@
 import requests
 import pprint
 
+import ray._private.runtime_env.constants as ray_constants
 from ray._private.utils import get_master_wheel_url, get_release_wheel_url
 
 
@@ -40,7 +41,7 @@ def update_progress(result):
 
     retry = set()
     for sys_platform in ["darwin", "linux", "win32"]:
-        for py_version in ["36", "37", "38", "39", "310"]:
+        for py_version in ray_constants.RUNTIME_ENV_CONDA_PY_VERSIONS:
             if "dev" in ray.__version__:
                 url = get_master_wheel_url(
                     ray_commit=ray.__commit__,