fix ruff pyright

Author: asheshvidyut

src/mcp/client/session_group.py

@@ -276,9 +276,7 @@ async def _establish_session(
             await session_stack.aclose()
             raise
 
-    async def _aggregate_components(
-        self, server_info: types.Implementation, session: mcp.ClientSession
-    ) -> None:
+    async def _aggregate_components(self, server_info: types.Implementation, session: mcp.ClientSession) -> None:
         """Aggregates prompts, resources, and tools from a given session."""
 
         # Create a reverse index so we can find all prompts, resources, and

src/mcp/client/transport_session.py

@@ -109,9 +109,6 @@ async def complete(
         """Send a completion/complete request."""
         raise NotImplementedError
 
-
-        
-
     @abstractmethod
     async def list_tools(
         self,

src/mcp/shared/context.py

@@ -1,15 +1,17 @@
 from dataclasses import dataclass
-from typing import Any, Generic
+from typing import TYPE_CHECKING, Any, Generic
 
 from typing_extensions import TypeVar
 
-from mcp.client.transport_session import ClientTransportSession
-from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.session import BaseSession
 from mcp.types import RequestId, RequestParams
 
+if TYPE_CHECKING:
+    from mcp.client.session import ClientTransportSession
+    from mcp.server.session import ServerTransportSession
+
 SessionT = TypeVar(
-    "SessionT", bound=BaseSession[Any, Any, Any, Any, Any] | ClientTransportSession | ServerTransportSession
+    "SessionT", bound=BaseSession[Any, Any, Any, Any, Any] | "ClientTransportSession" | "ServerTransportSession"
 )
 LifespanContextT = TypeVar("LifespanContextT")
 RequestT = TypeVar("RequestT", default=Any)