@@ -51,6 +51,8 @@ class CustomTableModel(QtCore.QAbstractT
def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
return QtCore.QVariant(self.headers[section])
elif orientation == QtCore.Qt.Vertical and role == QtCore.Qt.DisplayRole:
return QtCore.QVariant(section)
return super().headerData(section, orientation, role)
def rowCount(self, parent=None):