Added torch and gym to the project.

2023-02-12 13:50:59 +01:00
parent 1298f32850
commit b1b0c30114
2 changed files with 248 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,6 +19,9 @@ python = "3.10.*"
 scipy = "^1.10.0"
 tqdm = "^4.64.1"
 jupyterlab = "^3.6.1"
+torchvision = "^0.14.1"
+torchaudio = "^0.13.1"
+gym = "^0.26.2"

 [tool.poetry.group.build.dependencies]
 blackcellmagic = "^0.0.3"