diff --git a/sdk/python/requirements/py3.10-ci-requirements.txt b/sdk/python/requirements/py3.10-ci-requirements.txt index f31e77d2ca..b720cb7f4c 100644 --- a/sdk/python/requirements/py3.10-ci-requirements.txt +++ b/sdk/python/requirements/py3.10-ci-requirements.txt @@ -917,7 +917,7 @@ starlette==0.41.3 # via fastapi substrait==0.23.0 # via ibis-substrait -sympy==1.13.1 +sympy==1.13.3 # via torch tabulate==0.9.0 # via feast (setup.py) @@ -953,7 +953,7 @@ toolz==0.12.1 # dask # ibis-framework # partd -torch==2.5.1 +torch==2.4.1 # via feast (setup.py) tornado==6.4.2 # via diff --git a/sdk/python/requirements/py3.11-ci-requirements.txt b/sdk/python/requirements/py3.11-ci-requirements.txt index b1070487f5..6c0879f66c 100644 --- a/sdk/python/requirements/py3.11-ci-requirements.txt +++ b/sdk/python/requirements/py3.11-ci-requirements.txt @@ -908,7 +908,7 @@ starlette==0.41.3 # via fastapi substrait==0.23.0 # via ibis-substrait -sympy==1.13.1 +sympy==1.13.3 # via torch tabulate==0.9.0 # via feast (setup.py) @@ -934,7 +934,7 @@ toolz==0.12.1 # dask # ibis-framework # partd -torch==2.5.1 +torch==2.4.1 # via feast (setup.py) tornado==6.4.2 # via diff --git a/sdk/python/requirements/py3.9-ci-requirements.txt b/sdk/python/requirements/py3.9-ci-requirements.txt index 241497dc00..3c8ed28969 100644 --- a/sdk/python/requirements/py3.9-ci-requirements.txt +++ b/sdk/python/requirements/py3.9-ci-requirements.txt @@ -925,7 +925,7 @@ starlette==0.41.3 # via fastapi substrait==0.23.0 # via ibis-substrait -sympy==1.13.1 +sympy==1.13.3 # via torch tabulate==0.9.0 # via feast (setup.py) @@ -961,7 +961,7 @@ toolz==0.12.1 # dask # ibis-framework # partd -torch==2.5.1 +torch==2.4.1 # via feast (setup.py) tornado==6.4.2 # via diff --git a/setup.py b/setup.py index 36361045be..a54ab27c82 100644 --- a/setup.py +++ b/setup.py @@ -158,7 +158,7 @@ MILVUS_REQUIRED = ["pymilvus"] -PYTORCH_REQUIRED = ["torch>=2.5.0"] +PYTORCH_REQUIRED = ["torch>=2.4.0", "torch<2.5.0"] CI_REQUIRED = ( [