Skip to content

Commit 5eaaea8

Browse files
fix: Add _registered_method to grpc ChannelStub (#614)
* fix: Add _registered_method to grpc ChannelStub * Fix format * Fix black
1 parent 8eaea7a commit 5eaaea8

File tree

1 file changed

+26
-4
lines changed

1 file changed

+26
-4
lines changed

google/api_core/grpc_helpers.py

+26-4
Original file line numberDiff line numberDiff line change
@@ -569,20 +569,42 @@ def __getattr__(self, key):
569569
except KeyError:
570570
raise AttributeError
571571

572-
def unary_unary(self, method, request_serializer=None, response_deserializer=None):
572+
def unary_unary(
573+
self,
574+
method,
575+
request_serializer=None,
576+
response_deserializer=None,
577+
_registered_method=False,
578+
):
573579
"""grpc.Channel.unary_unary implementation."""
574580
return self._stub_for_method(method)
575581

576-
def unary_stream(self, method, request_serializer=None, response_deserializer=None):
582+
def unary_stream(
583+
self,
584+
method,
585+
request_serializer=None,
586+
response_deserializer=None,
587+
_registered_method=False,
588+
):
577589
"""grpc.Channel.unary_stream implementation."""
578590
return self._stub_for_method(method)
579591

580-
def stream_unary(self, method, request_serializer=None, response_deserializer=None):
592+
def stream_unary(
593+
self,
594+
method,
595+
request_serializer=None,
596+
response_deserializer=None,
597+
_registered_method=False,
598+
):
581599
"""grpc.Channel.stream_unary implementation."""
582600
return self._stub_for_method(method)
583601

584602
def stream_stream(
585-
self, method, request_serializer=None, response_deserializer=None
603+
self,
604+
method,
605+
request_serializer=None,
606+
response_deserializer=None,
607+
_registered_method=False,
586608
):
587609
"""grpc.Channel.stream_stream implementation."""
588610
return self._stub_for_method(method)

0 commit comments

Comments
 (0)