Compatibility fixes for python3.11 and ci running.

Author: fleming79

src/async_kernel/__main__.py

@@ -36,11 +36,12 @@ def main(wait_exit_context=anyio.sleep_forever):
         help=f"Add a kernel spec. Default kernels: {list(map(str, KernelName))}.\n"
         + "Other kernels and options are permitted. See: `write_kernel_spec` for detail.",
     )
+    kernels = [] if not kernel_dir.exists() else [item.name for item in kernel_dir.iterdir() if item.is_dir()]
     parser.add_argument(
         "-r",
         "--remove",
         dest="remove",
-        help=f"remove existing kernel specs. Installed kernels: {[item.name for item in kernel_dir.iterdir() if item.is_dir()]}",
+        help=f"remove existing kernel specs. Installed kernels: {kernels}",
     )
 
     args, unknownargs = parser.parse_known_args()

src/async_kernel/caller.py

@@ -13,10 +13,11 @@
 import weakref
 from collections import deque
 from collections.abc import AsyncGenerator, Awaitable, Callable, Generator
-from typing import TYPE_CHECKING, Any, ClassVar, Literal, Self, cast, override
+from typing import TYPE_CHECKING, Any, ClassVar, Literal, Self, cast
 
 import anyio
 import sniffio
+from typing_extensions import override
 from zmq import Context, Socket, SocketType
 
 from async_kernel.typing import NoValue, T

src/async_kernel/compiler.py

@@ -8,9 +8,10 @@
 import os
 import pathlib
 import tempfile
-from typing import ClassVar, override
+from typing import ClassVar
 
 from IPython.core.compilerop import CachingCompiler
+from typing_extensions import override
 
 __all__ = ["XCachingCompiler", "murmur2_x86"]
 

src/async_kernel/iostream.py

@@ -5,7 +5,9 @@
 
 from io import TextIOBase
 from threading import Lock
-from typing import TYPE_CHECKING, override
+from typing import TYPE_CHECKING
+
+from typing_extensions import override
 
 if TYPE_CHECKING:
     from collections.abc import Callable

src/async_kernel/kernel.py

@@ -435,7 +435,7 @@ def get_execute_mode(job: Job[ExecuteContent]) -> ExecuteMode:
         if m := job["msg"]["content"].get("execute_mode"):
             # Respect an existing mode
             return ExecuteMode(m)
-        if (c := job["msg"]["content"]["code"].strip().split("\n")[0].strip()) in ExecuteMode:
+        if (c := job["msg"]["content"]["code"].strip().split("\n")[0].strip()) in iter(ExecuteMode):
             mode = ExecuteMode(c)
         else:
             mode = ExecuteMode.queue

tests/references.py

@@ -7,10 +7,10 @@
 
 import re
 import sys
-from typing import override
 
 from packaging.version import Version as PackingVersion
 from traitlets import Bool, Dict, Enum, HasTraits, Integer, List, TraitError, Unicode, observe
+from typing_extensions import override
 
 __all__ = ["Reference", "references"]
 

tests/test_kernelspec.py

@@ -14,9 +14,10 @@
 @pytest.mark.parametrize("kernel_name", list(KernelName))
 def test_write_kernel_spec(kernel_name: KernelName):
     path = write_kernel_spec(kernel_name=kernel_name)
-    for fname in RESOURCES.iterdir():
-        dst = path.joinpath(fname)
-        assert pathlib.Path(dst).exists()
+    if RESOURCES.exists():
+        for fname in RESOURCES.iterdir():
+            dst = path.joinpath(fname)
+            assert pathlib.Path(dst).exists()
     kernel_json = path.joinpath("kernel.json")
     assert kernel_json.exists()
     with kernel_json.open("r") as f:

tests/test_start_in_context.py

@@ -28,7 +28,7 @@ def anyio_backend(kernel_name: KernelName):
 
 async def test_start_kernel_in_context(anyio_backend, kernel_name):
     utils.clear_kernel()
-    if kernel_name is KernelName.asyncio_eager:
+    if kernel_name == "async-eager":
         loop = asyncio.get_running_loop()
         loop.set_task_factory(asyncio.eager_task_factory)
     try: