Python AIO: Match continuation typing on Interceptors (#26500)

3 years ago · a3a7e55367
parent 41d2db8cb0
commit a3a7e55367
1 changed files with 2 additions and 2 deletions
--- a/src/python/grpcio/grpc/aio/_interceptor.py
+++ b/src/python/grpcio/grpc/aio/_interceptor.py
@ -169,7 +169,7 @@ class StreamUnaryClientInterceptor(ClientInterceptor, metaclass=ABCMeta):
    async def intercept_stream_unary(
        self,
        continuation: Callable[[ClientCallDetails, RequestType],
-                               UnaryStreamCall],
+                               StreamUnaryCall],
        client_call_details: ClientCallDetails,
        request_iterator: RequestIterableType,
    ) -> StreamUnaryCall:
@ -210,7 +210,7 @@ class StreamStreamClientInterceptor(ClientInterceptor, metaclass=ABCMeta):
    async def intercept_stream_stream(
        self,
        continuation: Callable[[ClientCallDetails, RequestType],
-                               UnaryStreamCall],
+                               StreamStreamCall],
        client_call_details: ClientCallDetails,
        request_iterator: RequestIterableType,
    ) -> Union[ResponseIterableType, StreamStreamCall]: