Pin torch to 2.2.2

Signed-off-by: Tomasz Wrona <tomasz@cast.ai>

Author: iamhatesz

pyproject.toml

@@ -96,7 +96,7 @@ opentelemetry = ["prometheus_client", "psutil"]
 spark = ["pyspark>=3.0.0,<4"]
 trino = ["trino>=0.305.0,<0.400.0", "regex"]
 postgres = ["psycopg[binary,pool]>=3.0.0,<4"]
-pytorch = ["torch>=2.2.2", "torchvision>=0.17.2"]
+pytorch = ["torch==2.2.2", "torchvision>=0.17.2"]
 qdrant = ["qdrant-client>=1.12.0"]
 redis = [
     "redis>=4.2.2,<5",

sdk/python/requirements/py3.10-ci-requirements.txt

@@ -1104,7 +1104,7 @@ starlette==0.46.1
     # via fastapi
 substrait==0.23.0
     # via ibis-substrait
-sympy==1.13.1
+sympy==1.13.3
     # via torch
 tabulate==0.9.0
     # via
@@ -1147,14 +1147,14 @@ toolz==0.12.1
     #   dask
     #   ibis-framework
     #   partd
-torch==2.6.0
+torch==2.2.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models
     #   easyocr
     #   safetensors
     #   torchvision
-torchvision==0.21.0
+torchvision==0.17.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models

sdk/python/requirements/py3.11-ci-requirements.txt

@@ -1098,7 +1098,7 @@ starlette==0.46.1
     # via fastapi
 substrait==0.23.0
     # via ibis-substrait
-sympy==1.13.1
+sympy==1.13.3
     # via torch
 tabulate==0.9.0
     # via
@@ -1133,14 +1133,14 @@ toolz==0.12.1
     #   dask
     #   ibis-framework
     #   partd
-torch==2.6.0
+torch==2.2.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models
     #   easyocr
     #   safetensors
     #   torchvision
-torchvision==0.21.0
+torchvision==0.17.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models

sdk/python/requirements/py3.9-ci-requirements.txt

@@ -1117,7 +1117,7 @@ starlette==0.46.1
     # via fastapi
 substrait==0.23.0
     # via ibis-substrait
-sympy==1.13.1
+sympy==1.13.3
     # via torch
 tabulate==0.9.0
     # via
@@ -1160,14 +1160,14 @@ toolz==0.12.1
     #   dask
     #   ibis-framework
     #   partd
-torch==2.6.0
+torch==2.2.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models
     #   easyocr
     #   safetensors
     #   torchvision
-torchvision==0.21.0
+torchvision==0.17.2
     # via
     #   feast (setup.py)
     #   docling-ibm-models

setup.py

@@ -164,7 +164,7 @@
 MILVUS_REQUIRED = ["pymilvus"]
 
 TORCH_REQUIRED = [
-    "torch>=2.2.2",
+    "torch==2.2.2",
     "torchvision>=0.17.2",
 ]