Include dict comprehension (facebookresearch#168)

IntrovertInDisguise · Oct 6, 2022 · 25cffd5 · 25cffd5
1 parent 1bb81f5
commit 25cffd5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)