From 3ad4fec6467aa94487cd0bae768a795c42e82f13 Mon Sep 17 00:00:00 2001
From: Madhur Karampudi <vkarampudi@google.com>
Date: Wed, 23 Oct 2024 10:11:36 -0700
Subject: [PATCH] Update required packages for fairness indicators to TF 2.16,
 TFMA 0.47.0.dev, and TFDV 1.16.1.

PiperOrigin-RevId: 689010533
---
 setup.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index 0451434..50c093e 100644
--- a/setup.py
+++ b/setup.py
@@ -38,15 +38,14 @@ def select_constraint(default, nightly=None, git_master=None):
     return default
 
 REQUIRED_PACKAGES = [
-    'tensorflow>=2.15,<2.16',
+    'tensorflow>=2.16,<2.17',
     'tensorflow-hub>=0.16.1,<1.0.0',
     'tensorflow-data-validation' + select_constraint(
-        default='>=1.15.1,<2.0.0',
-        nightly='>=1.16.0.dev',
+        default='>=1.16.1,<2.0.0',
+        nightly='>=1.17.0.dev',
         git_master='@git+https://github.com/tensorflow/data-validation@master'),
     'tensorflow-model-analysis' + select_constraint(
-        default='>=0.46,<0.47',
-        nightly='>=0.47.0.dev',
+        default='>=0.47.1,<0.48.0',
         git_master='@git+https://github.com/tensorflow/model-analysis@master'),
     'witwidget>=1.4.4,<2',
     'protobuf>=3.20.3,<5',