diff --git a/mbrl/util/logger.py b/mbrl/util/logger.py index 0aeaa91c..455e4375 100644 --- a/mbrl/util/logger.py +++ b/mbrl/util/logger.py @@ -94,7 +94,7 @@ def dump(self, step: int, prefix: str, save: bool = True, color: str = "yellow") if len(self._meters) == 0: return if save: - data = dict([(key, meter.value()) for key, meter in self._meters.items()]) + data = {key: meter.value() for key, meter in self._meters.items()} data["step"] = step self._dump_to_csv(data) self._dump_to_console(data, prefix, color)