如何使用单击按钮删除另一个类上的小部件(PyQt)

pip install PyQt5

from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton, QWidget

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setGeometry(100, 100, 300, 200)
        self.setWindowTitle('Main Window')

        self.button = QPushButton('Delete Widget', self)
        self.button.setGeometry(100, 100, 100, 30)
        self.button.clicked.connect(self.deleteWidget)

        self.widget = CustomWidget(self)
        self.widget.setGeometry(50, 50, 200, 100)

    def deleteWidget(self):
        self.widget.deleteLater()

class CustomWidget(QWidget):
    def __init__(self, parent):
        super().__init__(parent)
        self.initUI()

    def initUI(self):
        self.setStyleSheet('background-color: yellow;')

if __name__ == '__main__':
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()