revert file

Author: Kludex

tests/client/test_list_roots_callback.py

@@ -3,7 +3,6 @@
 
 from mcp.client.session import ClientSession
 from mcp.server.fastmcp.server import Context
-from mcp.server.session import ServerSession
 from mcp.shared.context import RequestContext
 from mcp.shared.memory import (
     create_connected_server_and_client_session as create_session,
@@ -36,7 +35,7 @@ async def list_roots_callback(
         return callback_return
 
     @server.tool("test_list_roots")
-    async def test_list_roots(context: Context[ServerSession], message: str):  # type: ignore[reportUnknownMemberType]
+    async def test_list_roots(context: Context, message: str):  # type: ignore[reportUnknownMemberType]
         roots = await context.session.list_roots()
         assert roots == callback_return
         return True