Be more cautious about cell 'id'

lsst-sqre · Oct 29, 2024 · 6062e9a · 6062e9a
1 parent dfde0c6
commit 6062e9a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/mobu/services/business/notebookrunner.py b/src/mobu/services/business/notebookrunner.py
@@ -296,14 +296,11 @@ async def execute_code(self, session: JupyterLabSession) -> None:
 
             for cell in self.read_notebook(self._notebook):
                 code = "".join(cell["source"])
-                if "id" in cell:
-                    cell_id = f'`{cell["id"]}` (#{cell["_index"]})'
-                else:
-                    cell_id = f'#{cell["_index"]}'
+                cell_id = cell.get("id") or "<unknown cell ID>"
                 ctx = CodeContext(
                     notebook=self._notebook.name,
                     path=str(self._notebook),
-                    cell=cell["id"],
+                    cell=cell_id,
                     cell_number=f"#{cell['_index']}",
                     cell_source=code,
                 )