TEMP

Signed-off-by: Matteo Bettini <matbet@meta.com>

Author: matteobettini

torchrl/collectors/collectors.py

@@ -766,7 +766,9 @@ def iterator(self) -> Iterator[TensorDictBase]:
                     tensordict_out = self.postproc(tensordict_out)
                 if self._exclude_private_keys:
                     excluded_keys = []
-                    for key in tensordict_out.keys(True, True):
+                    # Temp
+                    for key in tensordict_out.keys():
+                        # for key in tensordict_out.keys(True, True):
                         if isinstance(key, Tuple):
                             leaf_key = key[-1]
                         else: