如何将pyqtgraph图像示例集成到一个类中？

import pyqtgraph as pg
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget
from pyqtgraph.Qt import QtCore

class GraphWidget(QWidget):
    def __init__(self, parent=None):
        super(GraphWidget, self).__init__(parent)
        
        # 创建一个布局管理器
        layout = QVBoxLayout(self)
        
        # 创建一个pyqtgraph的绘图部件
        self.plot_widget = pg.PlotWidget()
        
        # 将绘图部件添加到布局管理器中
        layout.addWidget(self.plot_widget)
        
        # 设置布局管理器为窗口的主布局
        self.setLayout(layout)

    def update_plot(self):
        # 清空绘图部件中的图像
        self.plot_widget.clear()
        
        # 创建一个曲线对象
        curve = pg.PlotCurveItem(x=[1, 2, 3, 4, 5], y=[1, 2, 3, 4, 5])
        
        # 将曲线对象添加到绘图部件中
        self.plot_widget.addItem(curve)

class MainWindow(QMainWindow):
    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)
        
        # 创建一个自定义的QWidget子类对象
        self.graph_widget = GraphWidget()
        
        # 设置窗口的中央部件为自定义的QWidget子类对象
        self.setCentralWidget(self.graph_widget)
        
        # 调用自定义QWidget子类对象中的更新绘图部件方法
        self.graph_widget.update_plot()

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