From 3ff5740deb6f810d48547ff77f9c035ef8cfdf30 Mon Sep 17 00:00:00 2001
From: Sherif Akoush <sa@seldon.io>
Date: Mon, 18 Nov 2024 10:54:49 +0000
Subject: [PATCH] add min torch and tf versions for python 3.12 support

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

diff --git a/pyproject.toml b/pyproject.toml
index d4451b348..b0925f52e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -111,7 +111,7 @@ filelock = "^3.13.1"
 httpx-sse = ">=0.3.1,<0.5.0"
 
 [tool.poetry.group.docker.dependencies]
-tensorflow = ">=2.16"
+tensorflow = "^2.16"
 
 [tool.poetry.group.all-runtimes]
 optional = true
@@ -135,7 +135,7 @@ requests-mock = "~1.11.0"
 types-requests = ">=2.28.11.5,<2.32.0.0"
 
 ## Dev dependencies from MLflow
-torch = "2.2.1"
+torch = "^2.4"
 pytorch-lightning = "2.4.0"
 torchmetrics = "1.5.0"
 torchvision = "0.17.1"