diff --git a/test_constraints.txt b/test_constraints.txt
index 131727aa28f..4caa86f2246 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