update torch min version in mlflow runtime to support python 2.12

SeldonIO · Nov 18, 2024 · 8304a74 · 8304a74
1 parent 3ff5740
commit 8304a74
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -136,10 +136,6 @@ types-requests = ">=2.28.11.5,<2.32.0.0"
 
 ## Dev dependencies from MLflow
 torch = "^2.4"
-pytorch-lightning = "2.4.0"
-torchmetrics = "1.5.0"
-torchvision = "0.17.1"
-mlflow = "2.17.0"
 
 ## Dev dependencies from HuggingFace
 transformers = ">=4.30,<5.0"

diff --git a/runtimes/mlflow/pyproject.toml b/runtimes/mlflow/pyproject.toml
@@ -14,11 +14,7 @@ mlflow = "*"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}
-torch = "2.2.1"
-pytorch-lightning = "2.4.0"
-torchmetrics = "1.5.2"
-torchvision = "0.17.1"
-mlflow = "2.17.2"
+torch = "^2.4"
 
 [build-system]
 requires = ["poetry-core"]