fix: mark running cells as failed+done on restart

benlubas · May 10, 2024 · 7f1c31d · 7f1c31d
1 parent df5ccef
commit 7f1c31d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/rplugin/python3/molten/moltenbuffer.py b/rplugin/python3/molten/moltenbuffer.py
@@ -93,10 +93,15 @@ def interrupt(self) -> None:
 
     def restart(self, delete_outputs: bool = False) -> None:
         if delete_outputs:
-            self.outputs = {}
+            self.clear_virt_outputs()
             self.clear_interface()
             self.clear_open_output_windows()
-            self.clear_virt_outputs()
+            self.outputs = {}
+        else:
+            for output in self.outputs.values():
+                if output.output.status == OutputStatus.RUNNING:
+                    output.output.status = OutputStatus.DONE
+                    output.output.success = False
 
         self.runtime.restart()