def headerData()

in mbrl/diagnostics/training_browser.py [0:0]


    def headerData(self, row, orientation, role):
        if orientation == Qt.Vertical and role == Qt.DisplayRole:
            return (
                f"Experiment #{row+1}"
                if (row < (self.rowCount() - 2))
                else ("X Label" if (row < (self.rowCount() - 1)) else "Y Label")
            )
        return None