chore: fix mypy linting for websockets (#828)

Author: kumaraditya303

playwright/_impl/_transport.py

@@ -24,6 +24,7 @@
 
 import websockets
 from pyee import AsyncIOEventEmitter
+from websockets.client import connect as websocket_connect
 
 from playwright._impl._api_types import Error
 from playwright._impl._helper import ParsedMessagePayload
@@ -185,7 +186,7 @@ async def wait_until_stopped(self) -> None:
 
     async def run(self) -> None:
         try:
-            self._connection = await websockets.connect(
+            self._connection = await websocket_connect(
                 self.ws_endpoint, extra_headers=self.headers
             )
         except Exception as exc: