Ctrl + Z should undo the cell changes first (Or should not undo the outputs)

Environment data

• VS Code version: 1.55.2
• Jupyter Extension version (available under the Extensions sidebar): v2021.5.745244803
• Python Extension version (available under the Extensions sidebar): v2021.4.765268190
• OS (Windows | Mac | Linux distro) and version: Windows 10
• Python and/or Anaconda version: N/A
• Type of virtual environment used (N/A | venv | virtualenv | conda | …): N/A
• Jupyter server running: Local | Remote | N/A

Expected behaviour

Ctrl + Z should undo changes in the cell if cell has been focused. And I’m also not sure should Ctrl+Z can undo the outputs? (IMO: Not)

Actual behaviour

Outputs will be undone rather than the cell.

Steps to reproduce:

[NOTE: Self-contained, minimal reproducing code samples are extremely helpful and will expedite addressing your issue]

1. Prepare a simple notebook:

2. Run the cell and type something in the cell before cell runs done:

3. Click the cell to make sure the cell has been focused:

4. Ctrl + Z, Outputs disappeared.

5. Ctrl + Z, Code undo.

Logs

Not important

[IanMatthewHuff]

I don’t believe there is a current workaround on this. We can discuss at our triage meeting (Thursday) about what we’d expect Ctrl-Z to do in this situation. If we have agreement from our team on this change I could help point you are the code area for making this fix. Our focus is really heavily on the new UI currently, so we’d be unlikely to fix it ourselves.

[Kingwl]

Thanks to the team!