Use CustomUnitIDColum

Author: alejoe91

spikeinterface_gui/mergeview.py

@@ -270,7 +270,7 @@ def _qt_make_layout(self):
 
     def _qt_refresh(self):
         from .myqt import QT
-        from .utils_qt import CustomItem, CustomItemMergeUnitID
+        from .utils_qt import CustomItem, CustomItemUnitID
 
         self.table.clear()
         self.table.setSortingEnabled(False)
@@ -298,7 +298,7 @@ def _qt_refresh(self):
                     pix = QT.QPixmap(16, 16)
                     pix.fill(color)
                     icon = QT.QIcon(pix)
-                    item = CustomItemMergeUnitID(name)
+                    item = CustomItemUnitID(name)
                     item.setData(QT.Qt.ItemDataRole.UserRole, unit_id)
                     item.setFlags(QT.Qt.ItemIsEnabled | QT.Qt.ItemIsSelectable)
                     self.table.setItem(r, c, item)

spikeinterface_gui/utils_qt.py

@@ -474,17 +474,6 @@ def __lt__(self, other):
         other_ind = self.unit_ids.index(other.text())
         return ind < other_ind
 
-class CustomItemMergeUnitID(QT.QTableWidgetItem):
-    # special case for merge view unit_id columns: sort numerically by the unit_id stored in UserRole
-    def __lt__(self, other):
-        self_data = self.data(QT.Qt.ItemDataRole.UserRole)
-        other_data = other.data(QT.Qt.ItemDataRole.UserRole)
-        if self_data is not None and other_data is not None:
-            try:
-                return float(self_data) < float(other_data)
-            except (TypeError, ValueError):
-                pass
-        return self.text() < other.text()
 
 class OrderableCheckItem(QT.QTableWidgetItem):
     # special case for checkbox