Skip to content

Commit 170c6f3

Browse files
authored
[Doc] Fix missing tensordict install for doc (#685)
1 parent 69bdeaf commit 170c6f3

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

setup.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,13 @@ def _main(argv):
202202
"build_ext": BuildExtension.with_options(no_python_abi_suffix=True),
203203
"clean": clean,
204204
},
205-
install_requires=[pytorch_package_dep, "numpy", "packaging", "cloudpickle"],
205+
install_requires=[
206+
pytorch_package_dep,
207+
"numpy",
208+
"packaging",
209+
"cloudpickle",
210+
"tensordict-nightly",
211+
],
206212
extras_require={
207213
"atari": [
208214
"gym<=0.24",

0 commit comments

Comments
 (0)