From 47a334ab93e8dbac981b4e110c83146a8c466709 Mon Sep 17 00:00:00 2001 From: Cindy Zhang Date: Fri, 8 Nov 2024 15:47:04 -0800 Subject: [PATCH 1/2] fix Signed-off-by: Cindy Zhang --- python/ray/serve/tests/test_batching.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/serve/tests/test_batching.py b/python/ray/serve/tests/test_batching.py index b1e8b76d3e85..e87dc9426c4b 100644 --- a/python/ray/serve/tests/test_batching.py +++ b/python/ray/serve/tests/test_batching.py @@ -51,7 +51,7 @@ async def __call__(self, request): # Set the max batch size. handle = serve.run(NoListReturned.bind()) - with pytest.raises(ray.exceptions.RayTaskError): + with pytest.raises(TypeError): assert handle.remote(1).result() From d1143fd6d38fa07adb777a77c017f0fb6efe99d7 Mon Sep 17 00:00:00 2001 From: Cindy Zhang Date: Fri, 8 Nov 2024 15:59:05 -0800 Subject: [PATCH 2/2] lint Signed-off-by: Cindy Zhang --- python/ray/serve/tests/test_batching.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/ray/serve/tests/test_batching.py b/python/ray/serve/tests/test_batching.py index e87dc9426c4b..1c8f0b02ba66 100644 --- a/python/ray/serve/tests/test_batching.py +++ b/python/ray/serve/tests/test_batching.py @@ -7,7 +7,6 @@ import requests from starlette.responses import StreamingResponse -import ray from ray import serve