Internal change

PiperOrigin-RevId: 643225735
tensorflow · Jun 18, 2024 · e49de35 · e49de35
1 parent f739337
commit e49de35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git 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
+Flask-session<0.6.0
diff --git 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',
   ]