def __init__()

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

• 51 lines of code
• 4 McCabe index (conditional complexity)

    def __init__(self, experiment_root):
        super(BasicTrainingResultsWindow, self).__init__()

        self.experiment_root = experiment_root
        if self.experiment_root[-1] != "/":
            self.experiment_root = self.experiment_root + "/"

        self.chart = Figure()
        self.axes = self.chart.add_subplot(111)
        self.graphWidget = FigureCanvas(self.chart)
        self.setCentralWidget(self.graphWidget)

        self.logYAxisCheckbox = QCheckBox("Log Scale (Y Axis)")
        self.logYAxisCheckbox.stateChanged.connect(self.onChangeScale)

        self.displayAsDistributionCheckbox = QCheckBox("Aggregate Results (Mean/Std)")
        self.displayAsDistributionCheckbox.setChecked(True)

        self.saveFigureButton = QPushButton("&Save Figure")
        self.saveFigureButton.clicked.connect(self.onSaveFigure)

        self.optionsToolBar = QToolBar(self)
        self.optionsToolBar.addWidget(self.saveFigureButton)
        self.optionsToolBar.addWidget(self.logYAxisCheckbox)
        self.optionsToolBar.addWidget(self.displayAsDistributionCheckbox)
        self.addToolBar(self.optionsToolBar)

        self.load_experiments()

        self.resultsWidget = QDockWidget("Experiments", self)
        self.experimentTable = ExperimentsModel(
            self.experiment_results, self.displayAsDistributionCheckbox, self
        )
        self.tableView = QTableView()
        self.tableView.setModel(self.experimentTable)
        self.tableView.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.tableView.selectionModel().selectionChanged.connect(
            self.onExperimentsSelectionChanged
        )
        for i in range(self.tableView.horizontalHeader().count()):
            self.tableView.horizontalHeader().setSectionResizeMode(
                i, QHeaderView.Stretch
            )
        self.resultsWidget.setWidget(self.tableView)
        self.addDockWidget(Qt.LeftDockWidgetArea, self.resultsWidget)

        self.graphLabelsWidget = QDockWidget("Graph Labels", self)
        self.labelTable = GraphLabels(self.experiment_names, self)
        self.labelView = QTableView()
        self.labelView.setModel(self.labelTable)
        for i in range(self.labelView.horizontalHeader().count()):
            self.labelView.horizontalHeader().setSectionResizeMode(
                i, QHeaderView.Stretch
            )
        self.graphLabelsWidget.setWidget(self.labelView)
        self.addDockWidget(Qt.LeftDockWidgetArea, self.graphLabelsWidget)

        self.labelTable.dataChanged.connect(self.onLabelChanged)
        self.displayAsDistributionCheckbox.stateChanged.connect(
            self.onChangeDisplayAsDistribution
        )