Fixed errors when running MagmaDelete

haininhhoang94 · Sep 16, 2022 · 0b7d61e · 0b7d61e
1 parent ce56a23
commit 0b7d61e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/rplugin/python3/magma/magmabuffer.py b/rplugin/python3/magma/magmabuffer.py
@@ -108,8 +108,9 @@ def reevaluate_cell(self) -> None:
 
     def _check_if_done_running(self) -> None:
         # TODO: refactor
-        is_idle = self.current_output is None or (
-            self.current_output is not None
+        is_idle = ( self.current_output is None
+            or not self.current_output in self.outputs
+        ) or ( self.current_output is not None
             and self.outputs[self.current_output].output.status
             == OutputStatus.DONE
         )
@@ -121,7 +122,7 @@ def tick(self) -> None:
         self._check_if_done_running()
 
         was_ready = self.runtime.is_ready()
-        if self.current_output is None:
+        if self.current_output is None or not self.current_output in self.outputs:
             did_stuff = self.runtime.tick(None)
         else:
             did_stuff = self.runtime.tick(
@@ -155,7 +156,7 @@ def clear_interface(self) -> None:
             -1,
         )
         # and self.nvim.funcs.winbufnr(self.display_window) != -1:
-        if self.selected_cell is not None:
+        if self.selected_cell is not None and self.selected_cell in self.outputs:
             self.outputs[self.selected_cell].clear_interface()
         self.canvas.clear()