diff --git a/poetry.lock b/poetry.lock index be4614c5e..5b625f7a7 100644 --- a/poetry.lock +++ b/poetry.lock @@ -6243,4 +6243,4 @@ testing = ["big-O", "flake8 (<5)", "jaraco.functools", "jaraco.itertools", "more [metadata] lock-version = "2.0" python-versions = "^3.10" -content-hash = "a983ce4406b471912d26814a93bc661f0c533dca8239ba0282791c7a03ed2b10" +content-hash = "78ab715b98aa752c854c500a220ad1db6567c16fa3a92f23c485bfc91da20f0d" diff --git a/pyproject.toml b/pyproject.toml index b7021ff01..0bc987c44 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,7 @@ scikit-learn = ">=0.24.2" scikit-image = ">=0.20.0" requests = "^2.31.0" torch = "~2.0" -torchvision = "~0.15" +torchvision = "^0.15" torchmetrics = ">=0.5.1" pytorch-lightning = "2.0" einops = ">=0.3.0"