From f10f97e01c2a34e877a43a122a23fc794b47bae3 Mon Sep 17 00:00:00 2001
From: vkarampudi <vkarampudi@google.com>
Date: Thu, 13 Jun 2024 21:48:35 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 643225735
---
 test_constraints.txt | 6 +-----
 tfx/dependencies.py  | 2 +-
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/test_constraints.txt b/test_constraints.txt
index 131727aa28..4caa86f224 100644
--- a/test_constraints.txt
+++ b/test_constraints.txt
@@ -9,8 +9,4 @@
 # dependencies.
 
 # TODO(b/321609768): Remove pinned Flask-session version after resolving the issue.
-Flask-session<0.6.0
-
-#TODO(b/329181965): Remove once we migrate TFX to 2.16.
-tensorflow<2.16
-tensorflow-text<2.16
\ No newline at end of file
+Flask-session<0.6.0
\ No newline at end of file
diff --git a/tfx/dependencies.py b/tfx/dependencies.py
index 84098ae7c0..3b48b0cd56 100644
--- a/tfx/dependencies.py
+++ b/tfx/dependencies.py
@@ -203,7 +203,7 @@ def make_extra_packages_tfdf():
   # Required for tfx/examples/penguin/penguin_utils_tfdf_experimental.py
   return [
       # NOTE: TFDF 1.0.1 is only compatible with TF 2.10.x.
-      'tensorflow-decision-forests>=1.0.1,<1.9',
+      'tensorflow-decision-forests>=1.0.1,<2',
   ]