diff --git a/labelme/labelDialog.py b/labelme/labelDialog.py index fc4301d603263ad9db52064ae6744cbaf928c759..1be12713b97e142c38c7025449aba35dbf4ed57a 100644 --- a/labelme/labelDialog.py +++ b/labelme/labelDialog.py @@ -36,9 +36,6 @@ from .lib import newIcon # - Calculate optimal position so as not to go out of screen area. -BB = QtWidgets.QDialogButtonBox - - class LabelQLineEdit(QtWidgets.QLineEdit): def setListWidget(self, list_widget): @@ -63,9 +60,13 @@ class LabelDialog(QtWidgets.QDialog): layout = QtWidgets.QVBoxLayout() layout.addWidget(self.edit) # buttons - self.buttonBox = bb = BB(BB.Ok | BB.Cancel, QtCore.Qt.Horizontal, self) - bb.button(BB.Ok).setIcon(newIcon('done')) - bb.button(BB.Cancel).setIcon(newIcon('undo')) + self.buttonBox = bb = QtWidgets.QDialogButtonBox( + QtWidgets.QDialogButtonBox.Ok | QtWidgets.QDialogButtonBox.Cancel, + QtCore.Qt.Horizontal, + self, + ) + bb.button(bb.Ok).setIcon(newIcon('done')) + bb.button(bb.Cancel).setIcon(newIcon('undo')) bb.accepted.connect(self.validate) bb.rejected.connect(self.reject) layout.addWidget(bb)