[Formatting] Fix F401 lint advisory (#134)

pytorch · Dec 29, 2022 · 170d878 · 170d878
1 parent 190ffe3
commit 170d878
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git 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
@@ -1 +1,6 @@
 from .tensorclass import is_tensorclass, tensorclass
+
+__all__ = [
+    "is_tensorclass",
+    "tensorclass",
+]