Jeremiah Lowin commited on
Commit
7fb0010
·
1 Parent(s): fa6e614

Improve type inference

Browse files
Files changed (1) hide show
  1. src/fastmcp/client/client.py +1 -3
src/fastmcp/client/client.py CHANGED
@@ -125,8 +125,6 @@ class Client(Generic[ClientTransportT]):
125
  instance = super().__new__(cls)
126
  return instance
127
 
128
- transport: ClientTransportT
129
-
130
  def __init__(
131
  self,
132
  transport: ClientTransportT
@@ -145,7 +143,7 @@ class Client(Generic[ClientTransportT]):
145
  timeout: datetime.timedelta | float | int | None = None,
146
  init_timeout: datetime.timedelta | float | int | None = None,
147
  ):
148
- self.transport = infer_transport(transport) # type: ignore
149
  self._session: ClientSession | None = None
150
  self._exit_stack: AsyncExitStack | None = None
151
  self._nesting_counter: int = 0
 
125
  instance = super().__new__(cls)
126
  return instance
127
 
 
 
128
  def __init__(
129
  self,
130
  transport: ClientTransportT
 
143
  timeout: datetime.timedelta | float | int | None = None,
144
  init_timeout: datetime.timedelta | float | int | None = None,
145
  ):
146
+ self.transport = cast(ClientTransportT, infer_transport(transport))
147
  self._session: ClientSession | None = None
148
  self._exit_stack: AsyncExitStack | None = None
149
  self._nesting_counter: int = 0