Fix typings

jupyter-server · Sep 26, 2023 · 1bc77c3 · 1bc77c3
1 parent b5748a2
commit 1bc77c3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/jupyter_server/services/events/handlers.py b/jupyter_server/services/events/handlers.py
@@ -44,7 +44,7 @@ async def get(self, *args, **kwargs):
         """Get an event socket."""
         self.pre_get()
         res = super().get(*args, **kwargs)
-        if res:
+        if res is not None:
             await res
 
     async def event_listener(self, logger: EventLogger, schema_id: str, data: dict) -> None:

diff --git a/jupyter_server/services/kernels/connection/channels.py b/jupyter_server/services/kernels/connection/channels.py
@@ -85,6 +85,8 @@ class ZMQChannelsWebsocketConnection(BaseKernelWebsocketConnection):
         ),
     )
 
+    websocket_handler = Instance(KernelWebsocketHandler)
+
     @property
     def write_message(self):
         """Alias to the websocket handler's write_message method."""