Update

Vincent Moens · Vincent Moens · commit 5a59f00d758e · 2024-11-18T18:17:27.000Z
[ghstack-poisoned]
diff --git a/torchrl/data/map/tree.py b/torchrl/data/map/tree.py
@@ -163,9 +163,6 @@ def vertices(
             if h in memo and not use_path:
                 continue
             memo.add(h)
-            r = tree.rollout
-            if r is not None:
-                r = r["next", "observation"]
             if use_path:
                 result[cur_path] = tree
             elif use_id:
