From 9e932b5085cb4509f9bafec6cb682312afb81135 Mon Sep 17 00:00:00 2001 From: Cristian Ferretti Date: Tue, 4 Jun 2024 12:59:46 -0400 Subject: [PATCH] Manual fix. --- 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 6a824f3308d..6ad9a6d0405 100644 --- a/py/client/pydeephaven/session.py +++ b/py/client/pydeephaven/session.py @@ -222,7 +222,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 @@ -233,7 +233,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