Import sync from pycrdt

jupyter-server · Sep 13, 2024 · 6357140 · 6357140
1 parent 35984d0
commit 6357140
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/plugins/yjs/fps_yjs/ywidgets/widgets.py b/plugins/yjs/fps_yjs/ywidgets/widgets.py
@@ -1,15 +1,16 @@
 import sys
 from typing import Any
 
-from pycrdt import TransactionEvent
 
 try:
-    from ypywidgets.utils import (  # type: ignore
+    import ypywidgets
+    from pycrdt import (
+        TransactionEvent,
         YMessageType,
         YSyncMessageType,
+        create_sync_message,
         create_update_message,
         process_sync_message,
-        sync,
     )
     ypywidgets_installed = True
 except ImportError:
@@ -41,8 +42,8 @@ def comm_open(self, msg, comm) -> None:
             self.comm = comm
             model = self.ydocs[f"{name}Model"]()
             self.widgets[comm_id] = {"model": model, "comm": comm}
-            msg = sync(model.ydoc)
-            comm.send(**msg)
+            msg = create_sync_message(model.ydoc)
+            comm.send(buffers=[msg])
 
         def comm_msg(self, msg) -> None:
             comm_id = msg["content"]["comm_id"]

diff --git a/tests/test_execute.py b/tests/test_execute.py
@@ -55,7 +55,6 @@ async def recv(self) -> bytes:
         return bytes(b)
 
 
-@pytest.mark.skip(reason="FIXME: caused by new ypywidgets version")
 @pytest.mark.asyncio
 @pytest.mark.parametrize("auth_mode", ("noauth",))
 async def test_execute(auth_mode, unused_tcp_port):