From 0d9996939eb340b6c710b9b818b709771394fd39 Mon Sep 17 00:00:00 2001 From: Cristian Ferretti <37232625+jcferretti@users.noreply.github.com> Date: Tue, 4 Jun 2024 16:33:06 -0400 Subject: [PATCH] Fix names of type annotations not matching between definition and use. (#5562) --- py/client/pydeephaven/session.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/client/pydeephaven/session.py b/py/client/pydeephaven/session.py index 879b7424d0f..6581306b167 100644 --- a/py/client/pydeephaven/session.py +++ b/py/client/pydeephaven/session.py @@ -252,7 +252,7 @@ def update_metadata(self, metadata: Iterable[Tuple[str, Union[str, bytes]]]) -> self._auth_header_value = v if isinstance(v, bytes) else v.encode('ascii') break - def wrap_rpc(self, stub_call: NotBidiRpc, *args, **kwargs) -> Any: + def wrap_rpc(self, stub_call: _NotBidiRpc, *args, **kwargs) -> Any: if 'metadata' in kwargs: raise DHError('Internal error: "metadata" in kwargs not supported in wrap_rpc.') kwargs["metadata"] = self.grpc_metadata @@ -263,7 +263,7 @@ def wrap_rpc(self, stub_call: NotBidiRpc, *args, **kwargs) -> Any: # Now block until we get the result (or an exception) return future.result() - def wrap_bidi_rpc(self, stub_call: BidiRpc, *args, **kwargs) -> Any: + def wrap_bidi_rpc(self, stub_call: _BidiRpc, *args, **kwargs) -> Any: if 'metadata' in kwargs: raise DHError('Internal error: "metadata" in kwargs not supported in wrap_bidi_rpc.') kwargs["metadata"] = self.grpc_metadata