amend

facebookresearch · Sep 3, 2024 · fd1131a · fd1131a
1 parent 4adc919
commit fd1131a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -715,7 +715,6 @@ def _get_excluded_keys(self, group: str):
         for other_group in self.group_map.keys():
             if other_group != group:
                 excluded_keys += [other_group, ("next", other_group)]
-        excluded_keys += ["info", (group, "info"), ("next", group, "info")]
         return excluded_keys
 
     def _optimizer_loop(self, group: str) -> TensorDictBase: