Explicitly depend on tf_keras>=2.

PiperOrigin-RevId: 688254521
tensorflow · Oct 21, 2024 · 439a37a · 439a37a
1 parent b02397e
commit 439a37a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -62,6 +62,7 @@ def _make_required_install_packages():
           nightly='>=1.16.0.dev',
           git_master='@git+https://github.com/tensorflow/metadata@master',
       ),
+      'tf_keras>=2',
       'tfx-bsl'
       + select_constraint(
           default='>=1.15.1,<1.16.0',