diff --git a/requirements.txt b/requirements.txt index fbe09f617..39ac62061 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,4 @@ orjson pettingzoo requests tabulate -websockets==12.0 +websockets==14.2 diff --git a/src/poke_env/ps_client/ps_client.py b/src/poke_env/ps_client/ps_client.py index 8199b7da6..119a33d8e 100644 --- a/src/poke_env/ps_client/ps_client.py +++ b/src/poke_env/ps_client/ps_client.py @@ -10,8 +10,8 @@ from typing import Any, List, Optional, Set import requests -import websockets.client as ws -from websockets.client import WebSocketClientProtocol +import websockets as ws +from websockets import ClientConnection from websockets.exceptions import ConnectionClosedOK from poke_env.concurrency import ( @@ -84,7 +84,7 @@ def __init__( self._logged_in: Event = create_in_poke_loop(Event) self._sending_lock = create_in_poke_loop(Lock) - self.websocket: WebSocketClientProtocol + self.websocket: ClientConnection self._logger: Logger = self._create_logger(log_level) if start_listening: diff --git a/unit_tests/ps_client/test_ps_client.py b/unit_tests/ps_client/test_ps_client.py index 1bca5bfb4..532e9bb4a 100644 --- a/unit_tests/ps_client/test_ps_client.py +++ b/unit_tests/ps_client/test_ps_client.py @@ -134,7 +134,7 @@ async def test_listen(handle_message_mock): semaphore = asyncio.Semaphore() - async def showdown_server_mock(websocket, path): + async def showdown_server_mock(websocket): semaphore.release() await websocket.ping() await websocket.send("error|test 1")