From ba184a2667db6b1cf4e62a8ceefc8cf00a36bf50 Mon Sep 17 00:00:00 2001
From: Henry Mao <1828968+calclavia@users.noreply.github.com>
Date: Wed, 12 Mar 2025 13:51:31 -0700
Subject: [PATCH] Ruff

---
 src/mcp/client/websocket.py | 5 +++--
 tests/shared/test_ws.py     | 1 -
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/mcp/client/websocket.py b/src/mcp/client/websocket.py
index 37b6085..b807370 100644
--- a/src/mcp/client/websocket.py
+++ b/src/mcp/client/websocket.py
@@ -4,9 +4,9 @@ from contextlib import asynccontextmanager
 from typing import AsyncGenerator
 
 import anyio
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from pydantic import ValidationError
 from websockets.asyncio.client import connect as ws_connect
-from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from websockets.typing import Subprotocol
 
 import mcp.types as types
@@ -59,7 +59,8 @@ async def websocket_client(url: str) -> AsyncGenerator[
 
         async def ws_writer():
             """
-            Reads JSON-RPC messages from write_stream_reader and sends them to the server.
+            Reads JSON-RPC messages from write_stream_reader and 
+            sends them to the server.
             """
             async with write_stream_reader:
                 async for message in write_stream_reader:
diff --git a/tests/shared/test_ws.py b/tests/shared/test_ws.py
index 5b2628a..3fec798 100644
--- a/tests/shared/test_ws.py
+++ b/tests/shared/test_ws.py
@@ -8,7 +8,6 @@ import pytest
 import uvicorn
 from pydantic import AnyUrl
 from starlette.applications import Starlette
-from starlette.requests import Request
 from starlette.routing import WebSocketRoute
 
 from mcp.client.session import ClientSession