Fix YBaseDoc import

jupyterlab · Apr 2, 2024 · a40da5b · a40da5b
1 parent acfec42
commit a40da5b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jupyter_collaboration/ychat.py b/jupyter_collaboration/ychat.py
@@ -8,10 +8,11 @@
 from typing import Any, Callable, List
 
 import jupyter_ydoc
+from jupyter_ydoc.ybasedoc import YBaseDoc
 from pycrdt import Array, Map
 
 
-class YChat(jupyter_ydoc.YBaseDoc):
+class YChat(YBaseDoc):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self._ydoc["content"] = self._ycontent = Map()