Pin torchrl deps to 0.2.0 (#2594)

malfet · web-flow · commit 0bcfce21a6d0 · 2023-10-05T13:37:42.000-07:00
As latests seems to be incompatible with latest tensordict release, see pytorch/rl#1606
diff --git a/requirements.txt b/requirements.txt
@@ -25,7 +25,7 @@ tensorboard
 jinja2==3.0.3
 pytorch-lightning
 torchx
-torchrl
+torchrl==0.2.0
 ax-platform
 nbformat>=4.2.0
 datasets
