in mbrl/diagnostics/training_browser.py [0:0]
def reloadExperiments(self, data):
self.data = []
for path in data:
if path.endswith(SOURCE):
config = yaml.load(
open(
path.replace("/{}".format(SOURCE), "/.hydra/config.yaml"), "r"
),
Loader=yaml.FullLoader,
)
elif path.endswith(MULTI_ROOT):
config = yaml.load(open(path, "r"), Loader=yaml.FullLoader)
entry = [config["algorithm"]["name"]]
entry = entry + [config["experiment"]]
entry = entry + [
config["env"] if "env" in config else config["overrides"]["env"]
]
entry = entry + [config["seed"]]
self.data.append(entry)
self.modelReset.emit()