[BugFix] Fix LazyStackedCompositeSpec and introducing `consolidate_…

…spec` (pytorch#1392) Signed-off-by: Matteo Bettini <matbet@meta.com> Co-authored-by: vmoens <vincentmoens@gmail.com>
ai4co · Jul 31, 2023 · 575a0a4 · 575a0a4
1 parent c693199
commit 575a0a4
Show file tree

Hide file tree

Showing 6 changed files with 1,043 additions and 117 deletions.
diff --git a/docs/source/reference/data.rst b/docs/source/reference/data.rst
@@ -215,6 +215,8 @@ Check the :obj:`torchrl.envs.utils.check_env_specs` method for a sanity check.
     OneHotDiscreteTensorSpec
     UnboundedContinuousTensorSpec
     UnboundedDiscreteTensorSpec
+    LazyStackedTensorSpec
+    LazyStackedCompositeSpec
 
 Reinforcement Learning From Human Feedback (RLHF)
 -------------------------------------------------
@@ -253,3 +255,5 @@ Utils
     :template: rl_template.rst
 
     MultiStep
+    consolidate_spec
+    check_no_exclusive_keys