From e313d82bb04cf14321fea2eb830f920a045577c8 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 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test_constraints.txt b/test_constraints.txt
index 131727aa28f..cf2da72eac4 100644
--- a/test_constraints.txt
+++ b/test_constraints.txt
@@ -11,6 +11,6 @@
 # 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
+#TODO(b/329181965): Remove once we migrate TFX to 2.17.
+tensorflow<2.17
 tensorflow-text<2.16
\ No newline at end of file