Skip to content

Commit 575a0a4

Browse files
[BugFix] Fix LazyStackedCompositeSpec and introducing consolidate_spec (#1392)
Signed-off-by: Matteo Bettini <[email protected]> Co-authored-by: vmoens <[email protected]>
1 parent c693199 commit 575a0a4

File tree

6 files changed

+1043
-117
lines changed

6 files changed

+1043
-117
lines changed

docs/source/reference/data.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,8 @@ Check the :obj:`torchrl.envs.utils.check_env_specs` method for a sanity check.
215215
OneHotDiscreteTensorSpec
216216
UnboundedContinuousTensorSpec
217217
UnboundedDiscreteTensorSpec
218+
LazyStackedTensorSpec
219+
LazyStackedCompositeSpec
218220

219221
Reinforcement Learning From Human Feedback (RLHF)
220222
-------------------------------------------------
@@ -253,3 +255,5 @@ Utils
253255
:template: rl_template.rst
254256

255257
MultiStep
258+
consolidate_spec
259+
check_no_exclusive_keys

0 commit comments

Comments
 (0)