Skip to content

Commit

Permalink
Fix names of type annotations not matching between definition and use. (
Browse files Browse the repository at this point in the history
  • Loading branch information
jcferretti committed Jun 4, 2024
1 parent fd08f4a commit 0d99969
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions py/client/pydeephaven/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 0d99969

Please sign in to comment.