def load_experiments()

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

• 26 lines of code
• 6 McCabe index (conditional complexity)

    def load_experiments(self):
        self.experiment_results = glob.glob(
            self.experiment_root + "**/{}".format(SOURCE), recursive=True
        )
        self.multirun_results = glob.glob(
            self.experiment_root + "**/{}".format(MULTI_ROOT), recursive=True
        )
        if self.displayAsDistributionCheckbox.checkState():
            for path in self.multirun_results:
                root = copy.deepcopy(path).replace("/" + MULTI_ROOT, "")

                self.experiment_results = list(
                    filter(lambda entry: root not in entry, self.experiment_results)
                )

                self.experiment_results.append(path)

        self.experiment_names = []
        for path in self.experiment_results:
            name = (
                path.replace(self.experiment_root, "")
                .replace("/{}".format(SOURCE), "")
                .replace("/{}".format(MULTI_ROOT), "")
            )
            self.experiment_names.append(name)

        global user_name_dict
        for name in self.experiment_names:
            if name not in user_name_dict:
                user_name_dict[name] = name