Support S3 pre-signed url for .whl file

Signed-off-by: Jeffrey(Xilang) Yan <[email protected]>

python/ray/_private/runtime_env/packaging.py

@@ -233,11 +233,11 @@ def parse_uri(pkg_uri: str) -> Tuple[Protocol, str]:
         )
 
     if protocol in Protocol.remote_protocols():
-        if pkg_uri.endswith(".whl"):
+        if uri.path.endswith(".whl"):
             # Don't modify the .whl filename. See
             # https://peps.python.org/pep-0427/#file-name-convention
             # for more information.
-            package_name = pkg_uri.split("/")[-1]
+            package_name = uri.path.split("/")[-1]
         else:
             package_name = f"{protocol.value}_{uri.netloc}{uri.path}"
 

python/ray/tests/test_runtime_env_packaging.py

@@ -492,6 +492,11 @@ class TestParseUri:
             ("s3://bucket/file.zip", Protocol.S3, "s3_bucket_file.zip"),
             ("https://test.com/file.zip", Protocol.HTTPS, "https_test_com_file.zip"),
             ("gs://bucket/file.zip", Protocol.GS, "gs_bucket_file.zip"),
+            (
+                    "https://test.com/package-0.0.1-py2.py3-none-any.whl?param=value",
+                    Protocol.HTTPS,
+                    "package-0.0.1-py2.py3-none-any.whl",
+            ),
         ],
     )
     def test_parsing_remote_basic(self, parsing_tuple):