From 2fef3acbaf0ecfd25e40a4239337727a5f57e1dc Mon Sep 17 00:00:00 2001 From: Xuan Wang Date: Wed, 14 Feb 2024 18:45:23 +0000 Subject: [PATCH 1/3] fix: Add _registered_method to grpc ChannelStub --- google/api_core/grpc_helpers.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/google/api_core/grpc_helpers.py b/google/api_core/grpc_helpers.py index 21c7315f..47fdbb03 100644 --- a/google/api_core/grpc_helpers.py +++ b/google/api_core/grpc_helpers.py @@ -569,20 +569,26 @@ def __getattr__(self, key): except KeyError: raise AttributeError - def unary_unary(self, method, request_serializer=None, response_deserializer=None): + def unary_unary( + self, method, request_serializer=None, response_deserializer=None,_registered_method=False + ): """grpc.Channel.unary_unary implementation.""" return self._stub_for_method(method) - def unary_stream(self, method, request_serializer=None, response_deserializer=None): + def unary_stream( + self, method, request_serializer=None, response_deserializer=None, _registered_method=False + ): """grpc.Channel.unary_stream implementation.""" return self._stub_for_method(method) - def stream_unary(self, method, request_serializer=None, response_deserializer=None): + def stream_unary( + self, method, request_serializer=None, response_deserializer=None, _registered_method=False + ): """grpc.Channel.stream_unary implementation.""" return self._stub_for_method(method) def stream_stream( - self, method, request_serializer=None, response_deserializer=None + self, method, request_serializer=None, response_deserializer=None, _registered_method=False ): """grpc.Channel.stream_stream implementation.""" return self._stub_for_method(method) From dd917590e572136fff3b550ff5b3224650e493bc Mon Sep 17 00:00:00 2001 From: Xuan Wang Date: Wed, 14 Feb 2024 18:55:28 +0000 Subject: [PATCH 2/3] Fix format --- google/api_core/grpc_helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/api_core/grpc_helpers.py b/google/api_core/grpc_helpers.py index 47fdbb03..cbc26ab4 100644 --- a/google/api_core/grpc_helpers.py +++ b/google/api_core/grpc_helpers.py @@ -570,7 +570,7 @@ def __getattr__(self, key): raise AttributeError def unary_unary( - self, method, request_serializer=None, response_deserializer=None,_registered_method=False + self, method, request_serializer=None, response_deserializer=None, _registered_method=False ): """grpc.Channel.unary_unary implementation.""" return self._stub_for_method(method) From 079438d7586183c2142ba78dafcb82bba9e077b4 Mon Sep 17 00:00:00 2001 From: Xuan Wang Date: Wed, 14 Feb 2024 19:02:44 +0000 Subject: [PATCH 3/3] Fix black --- google/api_core/grpc_helpers.py | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) diff --git a/google/api_core/grpc_helpers.py b/google/api_core/grpc_helpers.py index cbc26ab4..1dcbb8b9 100644 --- a/google/api_core/grpc_helpers.py +++ b/google/api_core/grpc_helpers.py @@ -570,25 +570,41 @@ def __getattr__(self, key): raise AttributeError def unary_unary( - self, method, request_serializer=None, response_deserializer=None, _registered_method=False + self, + method, + request_serializer=None, + response_deserializer=None, + _registered_method=False, ): """grpc.Channel.unary_unary implementation.""" return self._stub_for_method(method) def unary_stream( - self, method, request_serializer=None, response_deserializer=None, _registered_method=False + self, + method, + request_serializer=None, + response_deserializer=None, + _registered_method=False, ): """grpc.Channel.unary_stream implementation.""" return self._stub_for_method(method) def stream_unary( - self, method, request_serializer=None, response_deserializer=None, _registered_method=False + self, + method, + request_serializer=None, + response_deserializer=None, + _registered_method=False, ): """grpc.Channel.stream_unary implementation.""" return self._stub_for_method(method) def stream_stream( - self, method, request_serializer=None, response_deserializer=None, _registered_method=False + self, + method, + request_serializer=None, + response_deserializer=None, + _registered_method=False, ): """grpc.Channel.stream_stream implementation.""" return self._stub_for_method(method)