From 5848f1f8b59e1db64dec5c533663bc8c67a6fa8e Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 3 Feb 2025 17:06:53 +0000
Subject: [PATCH] [Doc] Add missing classes to doc

ghstack-source-id: 2e174577aa33cc8d69c0f423c90ea2e5ee0fdef6
Pull Request resolved: https://github.com/pytorch/tensordict/pull/1203
---
 docs/source/reference/nn.rst | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/docs/source/reference/nn.rst b/docs/source/reference/nn.rst
index 0be9b9883..9a5820d0a 100644
--- a/docs/source/reference/nn.rst
+++ b/docs/source/reference/nn.rst
@@ -194,6 +194,7 @@ to build distributions from network outputs and get summary statistics or sample
     TensorDictModuleBase
     TensorDictModule
     ProbabilisticTensorDictModule
+    ProbabilisticTensorDictSequential
     TensorDictSequential
     TensorDictModuleWrapper
     CudaGraphModule
@@ -257,6 +258,10 @@ Distributions
     NormalParamExtractor
     OneHotCategorical
     TruncatedNormal
+    InteractionType
+    set_interaction_type
+    add_custom_mapping
+    mappings
 
 
 Utils
@@ -270,8 +275,8 @@ Utils
 
     make_tensordict
     dispatch
-    set_interaction_type
     inv_softplus
     biased_softplus
     set_skip_existing
     skip_existing
+    rand_one_hot