diff --git a/src/colab_cli/runtime.py b/src/colab_cli/runtime.py index 5ebc262..e6cd94f 100644 --- a/src/colab_cli/runtime.py +++ b/src/colab_cli/runtime.py @@ -115,7 +115,12 @@ def kernel_client(self): }, ) else: - self._kernel_client = jupyter_kernel_client.KernelClient( + # jupyter-kernel-client >= 1.0.0 renamed KernelClient → + # JupyterKernelClient; fall back gracefully for both versions. + _KernelCls = getattr( + jupyter_kernel_client, "KernelClient", None + ) or jupyter_kernel_client.JupyterKernelClient + self._kernel_client = _KernelCls( server_url=self.url, token=self.token, kernel_id=self.kernel_id,