Skip to content

Commit

Permalink
Add cell execution_state (#197)
Browse files Browse the repository at this point in the history
  • Loading branch information
davidbrochart authored Aug 6, 2024
1 parent b8db63f commit 16162f9
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions jupyter_ydoc/ynotebook.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ class YNotebook(YBaseDoc):
"cell_type": str,
"source": YText,
"metadata": YMap,
"execution_state": str,
"execution_count": Int | None,
"outputs": [] | None,
"attachments": {} | None
Expand Down Expand Up @@ -100,6 +101,7 @@ def get_cell(self, index: int) -> Dict[str, Any]:
"""
meta = self._ymeta.to_py()
cell = self._ycells[index].to_py()
cell.pop("execution_state", None)
cast_all(cell, float, int) # cells coming from Yjs have e.g. execution_count as float
if "id" in cell and meta["nbformat"] == 4 and meta["nbformat_minor"] <= 4:
# strip cell IDs if we have notebook format 4.0-4.4
Expand Down Expand Up @@ -164,6 +166,7 @@ def create_ycell(self, value: Dict[str, Any]) -> Map:
output["text"] = Array(output.get("text", []))
outputs[idx] = Map(output)
cell["outputs"] = Array(outputs)
cell["execution_state"] = "idle"

return Map(cell)

Expand Down

0 comments on commit 16162f9

Please sign in to comment.