Remove restrict on keras

Signed-off-by: Simon Zhao <[email protected]>

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": [