From 04fc9afd33fcc1cb11a404a3505135d887378e32 Mon Sep 17 00:00:00 2001
From: Matteo Fordiani <matteofordiani@gmail.com>
Date: Mon, 13 Nov 2023 14:11:11 +0100
Subject: [PATCH] build(torch): relax torch constraints to allow for torch v2

Issue: #135
---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7e22fcfa..01565aa1 100644
--- 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"