[serve] small test change (ray-project#48665)

## Why are these changes needed? Match for the more specific error that is raised, in this case `TypeError`, instead of matching for generic `RayTaskError`. --------- Signed-off-by: Cindy Zhang <[email protected]>
JP-sDEV · Nov 14, 2024 · 893b901 · 893b901
1 parent a594e7d
commit 893b901
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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
 
 
@@ -51,7 +50,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()