From ad74d1b19e7552ce40b7a2af24fcbee9f8e830a6 Mon Sep 17 00:00:00 2001
From: anuartb <anuartb@google.com>
Date: Tue, 12 Mar 2024 07:53:00 -0700
Subject: [PATCH] Restrict some dependencies temporarily.

PiperOrigin-RevId: 615039730
---
 test_constraints.txt | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/test_constraints.txt b/test_constraints.txt
index 4caa86f224..131727aa28 100644
--- a/test_constraints.txt
+++ b/test_constraints.txt
@@ -9,4 +9,8 @@
 # dependencies.
 
 # TODO(b/321609768): Remove pinned Flask-session version after resolving the issue.
-Flask-session<0.6.0
\ No newline at end of file
+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