Manual fix.

deephaven · Jun 4, 2024 · 9e932b5 · 9e932b5
1 parent 108ec14
commit 9e932b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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