chore: fix mypy linting for websockets (microsoft#828)

kumaraditya303 · web-flow · commit f57cc8281422 · 2021-07-29T11:05:27.000+02:00
diff --git a/playwright/_impl/_transport.py b/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:
