[Doc] Fix missing tensordict install for doc (pytorch#685)

ai4co · Nov 17, 2022 · 170c6f3 · 170c6f3
1 parent 69bdeaf
commit 170c6f3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -202,7 +202,13 @@ def _main(argv):
             "build_ext": BuildExtension.with_options(no_python_abi_suffix=True),
             "clean": clean,
         },
-        install_requires=[pytorch_package_dep, "numpy", "packaging", "cloudpickle"],
+        install_requires=[
+            pytorch_package_dep,
+            "numpy",
+            "packaging",
+            "cloudpickle",
+            "tensordict-nightly",
+        ],
         extras_require={
             "atari": [
                 "gym<=0.24",