From e2d72e8c1de3780323958604663b6dfb4d1f57f5 Mon Sep 17 00:00:00 2001 From: Simon Zhao Date: Tue, 12 Nov 2024 12:50:15 +0800 Subject: [PATCH] Remove restrict on keras Signed-off-by: Simon Zhao --- setup.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/setup.py b/setup.py index 6fe5e62d6..670cd2dcc 100644 --- a/setup.py +++ b/setup.py @@ -34,7 +34,6 @@ "locust>=2.12.2,<3", # requires jinja2 "memory-profiler>=0.61.0,<1", "nltk>=3.8.1,<4", # requires tqdm - "numpy>=1.26.4;python_version>='3.12'", # https://stackoverflow.com/a/77364602/4505998 "notebook>=6.5.5,<8", # requires ipykernel, jinja2, jupyter, nbconvert, nbformat, packaging, requests "numba>=0.57.0,<1", "pandas>2.0.0,<3.0.0", # requires numpy @@ -56,12 +55,8 @@ "numpy<1.25.0;python_version<='3.8'", "nvidia-ml-py>=11.525.84", "spacy<=3.7.5;python_version<='3.8'", - "tensorflow>=2.8.4,!=2.9.0.*,!=2.9.1,!=2.9.2,!=2.10.0.*,<2.16; python_version<='3.8'", # Fixed TF due to constant security problems and breaking changes #2073 - "tensorflow~=2.16; python_version>'3.8'", # Version needed for python 3.12 + "tensorflow>=2.8.4,!=2.9.0.*,!=2.9.1,!=2.9.2,!=2.10.0.*,<2.16", # Fixed TF due to constant security problems and breaking changes #2073 "tf-slim>=1.1.0", # No python_requires in its setup.py - # Use keras 2 instead of keras 3 - "tf-keras~=2.15; python_version<='3.8'", # Not compatible with py3.12 - "tf-keras~=2.16; python_version>'3.8'", "torch>=2.0.1,<3", ], "spark": [