more typing cleanup

jupyter · May 5, 2022 · 914a384 · 914a384
1 parent abd2afb
commit 914a384
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/nbclient/tests/test_client.py b/nbclient/tests/test_client.py
@@ -206,8 +206,8 @@ def test_mock_wrapper(self):
             cell_mock = NotebookNode(
                 source='"foo" = "bar"', metadata={}, cell_type='code', outputs=[]
             )
-            executor = NotebookClient({})  # type:ignore
-            executor.nb = {'cells': [cell_mock]}  # type:ignore
+            executor = NotebookClient({})
+            executor.nb = {'cells': [cell_mock]}
 
             # self.kc.iopub_channel.get_msg => message_mock.side_effect[i]
             message_mock = iopub_messages_mock()
@@ -503,7 +503,7 @@ class TestExecute(NBClientTestsBase):
     maxDiff = None
 
     def test_constructor(self):
-        NotebookClient({})  # type:ignore
+        NotebookClient({})
 
     def test_populate_language_info(self):
         nb = nbformat.v4.new_notebook()  # Certainly has no language_info.

diff --git a/nbclient/util.py b/nbclient/util.py
@@ -65,7 +65,7 @@ def just_run(coro: Awaitable) -> Any:
 
 T = TypeVar("T")
 
-def run_sync(coro: Callable[..., T]) -> Callable[..., T]:
+def run_sync(coro: Callable[..., Awaitable[T]]) -> Callable[..., T]:
     """Runs a coroutine and blocks until it has executed.
 
     An event loop is created if no one already exists. If an event loop is