Skip to content

Commit 1d47163

Browse files
authored
Fix usage of await (#106)
* Fix usage of await * Fix types
1 parent d289632 commit 1d47163

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

jupyter_server_terminals/handlers.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import typing as t
77

8+
from jupyter_core.utils import ensure_async
89
from jupyter_server._tz import utcnow
910
from jupyter_server.auth.utils import warn_disabled_authorization
1011
from jupyter_server.base.handlers import JupyterHandler
@@ -54,11 +55,11 @@ async def get(self, *args: t.Any, **kwargs: t.Any) -> None:
5455
raise web.HTTPError(404)
5556
resp = super().get(*args, **kwargs)
5657
if resp is not None:
57-
await resp
58+
await ensure_async(resp) # type:ignore[arg-type]
5859

5960
async def on_message(self, message: t.Any) -> None: # type:ignore[override]
6061
"""Handle a socket message."""
61-
await super().on_message(message)
62+
await ensure_async(super().on_message(message)) # type:ignore[arg-type]
6263
self._update_activity()
6364

6465
def write_message(self, message: t.Any, binary: bool = False) -> None: # type:ignore[override]

0 commit comments

Comments
 (0)