fix: updated _delete_single_drive method to update titles

jonescompneurolab · Sep 23, 2024 · eb212bc · eb212bc
1 parent 13b1a52
commit eb212bc
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/hnn_core/gui/gui.py b/hnn_core/gui/gui.py
@@ -640,15 +640,22 @@ def _cell_layer_radio_change(value):
                                               'value')
 
     def _delete_single_drive(self, b):
+        index = self.drive_accordion.selected_index
+
         # Remove selected drive from drive lists
-        self.drive_boxes.pop(self.drive_accordion.selected_index)
-        self.drive_widgets.pop(self.drive_accordion.selected_index)
+        self.drive_boxes.pop(index)
+        self.drive_widgets.pop(index)
 
         # Rebuild the accordion collection
+        self.drive_accordion.titles = tuple(
+            t for i, t in enumerate(self.drive_accordion.titles) if i != index
+        )
+        self.drive_accordion.selected_index = None
+        self.drive_accordion.children = self.drive_boxes
+
+        # Render
         self._drives_out.clear_output()
         with self._drives_out:
-            self.drive_accordion.selected_index = None
-            self.drive_accordion.children = self.drive_boxes
             display(self.drive_accordion)
 
     def compose(self, return_layout=True):