TEMP

Signed-off-by: Matteo Bettini <[email protected]>
matteobettini · Aug 18, 2023 · e8e410e · e8e410e
1 parent 451e9a9
commit e8e410e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchrl/collectors/collectors.py b/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: