diff --git a/playwright/_impl/_connection.py b/playwright/_impl/_connection.py index de5ecd572..1c948a421 100644 --- a/playwright/_impl/_connection.py +++ b/playwright/_impl/_connection.py @@ -19,7 +19,12 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Union from greenlet import greenlet -from pyee import AsyncIOEventEmitter + +try: + from pyee.asyncio import AsyncIOEventEmitter +except ImportError: + # pyee < 9.0.0 + from pyee import AsyncIOEventEmitter from playwright._impl._helper import ParsedMessagePayload, parse_error from playwright._impl._transport import Transport diff --git a/playwright/_impl/_transport.py b/playwright/_impl/_transport.py index 9c26e4f06..4f909750f 100644 --- a/playwright/_impl/_transport.py +++ b/playwright/_impl/_transport.py @@ -24,7 +24,13 @@ import websockets import websockets.exceptions -from pyee import AsyncIOEventEmitter + +try: + from pyee.asyncio import AsyncIOEventEmitter +except ImportError: + # pyee < 9.0.0 + from pyee import AsyncIOEventEmitter + from websockets.client import connect as websocket_connect from playwright._impl._api_types import Error