build(torch): relax torch constraints to allow for torch v2

Issue: #135
autonlab · Nov 13, 2023 · 04fc9af · 04fc9af
1 parent 5dde465
commit 04fc9af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,12 +8,12 @@ packages = [{include = "auton_survival"}]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-torch = "^1.13"
+torch = ">=1.13"
 numpy = "^1.24"
 pandas = "^1.5"
 tqdm = "^4.66"
 scikit-learn = "^1.2"
-torchvision = "^0.14"
+torchvision = ">=0.14"
 scikit-survival = "^0.21"
 lifelines = "^0.26"