Adds dependency for Scikit-learn examples.

PiperOrigin-RevId: 360312375
speg03 · Mar 2, 2021 · 9155e50 · 9155e50
1 parent d86b4c5
commit 9155e50
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tfx/dependencies.py b/tfx/dependencies.py
@@ -169,6 +169,8 @@ def make_extra_packages_examples():
       # Required for tfx/examples/ranking
       'tensorflow-ranking>=0.3.3,<0.4',
       'struct2tensor>=0.28,<0.29',
+      # Required for tfx/examples/penguin/experimental
+      'scikit-learn>=0.24,<0.25',
   ]