[Setup] Add torchrl dependency

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Oct 6, 2023 · ee440e0 · ee440e0
1 parent b6c0a79
commit ee440e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
     author="Matteo Bettini",
     author_email="[email protected]",
     packages=find_packages(),
-    install_requires=["tqdm", "hydra-core"],
+    install_requires=["torchrl>=0.2.0", "tqdm", "hydra-core"],
     extras_require={
         "vmas": ["vmas>=1.2.10"],
         "pettingzoo": ["pettingzoo[all]>=1.24.1"],