Probably fix CLI

Author: flying-sheep

src/prelude_runner/cli.py

@@ -1,8 +1,10 @@
 from argparse import ArgumentParser
 from pathlib import Path
 from typing import Protocol
+import nbformat
 
 from .core import Preludes, execute
+from .types import Notebook
 
 
 def load_preludes(d: Path) -> Preludes:
@@ -27,5 +29,6 @@ def parse_args(argv: list[str] | None = None) -> Args:
 def main(argv: list[str] | None = None) -> None:
     args = parse_args(argv)
     preludes = load_preludes(args.preludes)
-    for nb in args.nb_path.rglob("*.ipynb"):
+    for nb_path in args.nb_path.rglob("*.ipynb"):
+        nb: Notebook = nbformat.reads(nb_path.read_text(), 4)
         execute(nb, cwd=args.nb_path, preludes=preludes)

src/prelude_runner/core.py

@@ -5,7 +5,6 @@
 from jupyter_client.manager import KernelManager
 from nbclient import NotebookClient
 from nbclient.util import ensure_async
-from nbformat import NotebookNode
 
 from .types import CodeCell, Notebook
 
@@ -35,13 +34,13 @@ async def on_cell_execute(cell: CodeCell, cell_index: int) -> None:
 
 
 def execute(
-    nb: NotebookNode,
+    nb: Notebook,
     preludes: Preludes,
     *,
     cwd: Path | None = None,
     km: KernelManager | None = None,
     **kwargs: Any,
-) -> NotebookNode:
+) -> Notebook:
     """Execute a notebook's code, updating outputs within the notebook object."""
     resources = {}
     if cwd is not None: