Add torch to pyproject.toml

bhilbert4 · Jan 17, 2025 · 4a35eac · 4a35eac
1 parent 6802e2d
commit 4a35eac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,6 +43,8 @@ dependencies = [
     "scipy>=1.12.0,<2",
     "selenium>=4.18.1,<5",
     "sqlalchemy>=2.0.29,<3",
+    "torch>=2.2.2,<2.5.2",
+    "torchvision>=0.17.2,<0.20.2",
     "wtforms>=3.1.2,<4",
 ]
 dynamic = [