Set selected instance to None after removal

talmolab · Jun 13, 2024 · 9b4933a · 9b4933a
1 parent 5c4c5db
commit 9b4933a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sleap/gui/commands.py b/sleap/gui/commands.py
@@ -2296,6 +2296,8 @@ def _do_deletion(context: CommandContext, lf_inst_list: List[int]):
         lfs_to_remove = []
         for lf, inst in lf_inst_list:
             context.labels.remove_instance(lf, inst, in_transaction=True)
+            if context.state["instance"] == inst:
+                context.state["instance"] = None
             if len(lf.instances) == 0:
                 lfs_to_remove.append(lf)
 
@@ -2533,6 +2535,7 @@ def do_action(context: CommandContext, params: dict):
             return
 
         context.labels.remove_instance(context.state["labeled_frame"], selected_inst)
+        context.state["instance"] = None
 
 
 class DeleteSelectedInstanceTrack(EditCommand):
@@ -2550,6 +2553,7 @@ def do_action(context: CommandContext, params: dict):
 
         track = selected_inst.track
         context.labels.remove_instance(context.state["labeled_frame"], selected_inst)
+        context.state["instance"] = None
 
         if track is not None:
             # remove any instance on this track