From 170d878de4d74a88a631f16df4d4d4fe0ddacb2f Mon Sep 17 00:00:00 2001
From: Alessandro Pietro Bardelli <apbard@users.noreply.github.com>
Date: Thu, 29 Dec 2022 07:47:03 +0100
Subject: [PATCH] [Formatting] Fix F401 lint advisory (#134)

---
 tensordict/nn/__init__.py        | 13 +++++++++++++
 tensordict/prototype/__init__.py |  5 +++++
 2 files changed, 18 insertions(+)

diff --git a/tensordict/nn/__init__.py b/tensordict/nn/__init__.py
index d1ce6e580..814f21533 100644
--- a/tensordict/nn/__init__.py
+++ b/tensordict/nn/__init__.py
@@ -11,3 +11,16 @@
     set_interaction_mode,
 )
 from .sequence import TensorDictSequential
+
+__all__ = [
+    "dispatch_kwargs",
+    "TensorDictModule",
+    "TensorDictModuleWrapper",
+    "get_functional",
+    "make_functional",
+    "repopulate_module",
+    "ProbabilisticTensorDictModule",
+    "ProbabilisticTensorDictSequential",
+    "set_interaction_mode",
+    "TensorDictSequential",
+]
diff --git a/tensordict/prototype/__init__.py b/tensordict/prototype/__init__.py
index cd752d6a0..3752c15c5 100644
--- a/tensordict/prototype/__init__.py
+++ b/tensordict/prototype/__init__.py
@@ -1 +1,6 @@
 from .tensorclass import is_tensorclass, tensorclass
+
+__all__ = [
+    "is_tensorclass",
+    "tensorclass",
+]