Skip to content

Commit 0239765

Browse files
authored
Fix websocket provider import in tests
1 parent c8e75dd commit 0239765

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/test_pycrdt_yjs.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
import pytest
88
from anyio import Event, create_task_group, move_on_after
99
from httpx_ws import aconnect_ws
10-
from pycrdt import Doc, Map
11-
from pycrdt.websocket import WebsocketProvider
10+
from pycrdt import Doc, Map, Provider
1211
from utils import Websocket
1312

1413
from jupyter_ydoc import YNotebook
@@ -66,7 +65,7 @@ async def test_ypy_yjs_0(yws_server, yjs_client):
6665
ydoc = Doc()
6766
ynotebook = YNotebook(ydoc)
6867
room_name = "my-roomname"
69-
async with aconnect_ws(f"http://localhost:{port}/{room_name}") as websocket, WebsocketProvider(
68+
async with aconnect_ws(f"http://localhost:{port}/{room_name}") as websocket, Provider(
7069
ydoc, Websocket(websocket, room_name)
7170
):
7271
nb = stringify_source(json.loads((files_dir / "nb0.ipynb").read_text()))

0 commit comments

Comments
 (0)