From 445805d971fb5f7681787aa910d36578a7451bcb Mon Sep 17 00:00:00 2001
From: Responsible ML Infra <responsible-ml-infra@google.com>
Date: Tue, 16 Jan 2024 10:35:18 -0800
Subject: [PATCH] Fix package dependencies to make it installable

PiperOrigin-RevId: 598885235
---
 setup.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index b1677a8..19bd9d4 100644
--- a/setup.py
+++ b/setup.py
@@ -38,14 +38,14 @@ def select_constraint(default, nightly=None, git_master=None):
     return default
 
 REQUIRED_PACKAGES = [
-    'tensorflow>=2.12.0,<2.13',
+    'tensorflow>=2.15,<3',
     'tensorflow-hub>=0.8.0,<1.0.0',
     'tensorflow-data-validation' + select_constraint(
-        default='>=1.13.0,<1.14.0',
+        default='>=1.14.0',
         nightly='>=1.14.0.dev',
         git_master='@git+https://github.com/tensorflow/data-validation@master'),
     'tensorflow-model-analysis' + select_constraint(
-        default='>=0.44,<0.45',
+        default='>=0.45',
         nightly='>=0.45.0.dev',
         git_master='@git+https://github.com/tensorflow/model-analysis@master'),
     'witwidget>=1.4.4,<2',
@@ -74,7 +74,7 @@ def select_constraint(default, nightly=None, git_master=None):
     package_data={
         'fairness_indicators': ['documentation/*'],
     },
-    python_requires='>=3.8,<3.10',
+    python_requires='>=3.9,<4',
     install_requires=REQUIRED_PACKAGES,
     tests_require=REQUIRED_PACKAGES,
     # PyPI package information.