Remove torchrl/tensordict from dependecies + update poetry cpu

2024-04-09 03:50:49 +00:00
parent 253e495df2
commit 19e7661b8d
4 changed files with 72 additions and 126 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,8 +39,6 @@ scikit-image = "^0.22.0"
 numba = "^0.59.0"
 mpmath = "^1.3.0"
 torch = "^2.2.1"
-tensordict = {git = "https://github.com/pytorch/tensordict"}
-torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"}
 mujoco = "^2.3.7"
 opencv-python = "^4.9.0.80"
 diffusers = "^0.26.3"