diff --git a/setup.py b/setup.py index e45c211b58..8d7899db6c 100644 --- a/setup.py +++ b/setup.py @@ -33,19 +33,19 @@ exec(fp.read(), version) version = version["__version__"] -tensorboard_extra_require = ["tensorflow >=2.3.0, <=2.7.0"] +tensorboard_extra_require = ["tensorflow >=2.3.0, <3.0.0dev"] metadata_extra_require = ["pandas >= 1.0.0"] -xai_extra_require = ["tensorflow >=2.3.0, <=2.5.0"] +xai_extra_require = ["tensorflow >=2.3.0, <3.0.0dev"] lit_extra_require = [ - "tensorflow >= 2.3.0", + "tensorflow >= 2.3.0, <3.0.0dev", "pandas >= 1.0.0", "lit-nlp >= 0.4.0", "explainable-ai-sdk >= 1.0.0", ] profiler_extra_require = [ - "tensorboard-plugin-profile >= 2.4.0", + "tensorboard-plugin-profile >= 2.4.0, <3.0.0dev", "werkzeug >= 2.0.0", - "tensorflow >=2.4.0", + "tensorflow >=2.4.0, <3.0.0dev", ] featurestore_extra_require = [ "google-cloud-bigquery-storage",