Update scikit learn dependency version

PiperOrigin-RevId: 276176947

Author: paulgc17

RELEASE.md

@@ -20,6 +20,7 @@
 * Building TFDV from source does not need pyarrow anymore.
 * Depends on `apache-beam[gcp]>=2.16,<3`.
 * Depends on `six>=1.12,<2`.
+* Depends on `scikit-learn>=0.18,<0.22`.
 * Depends on `tfx-bsl>=0.15,<0.16`.
 * Depends on `tensorflow-metadata>=0.15,<0.16`.
 * Depends on `tensorflow-transform>=0.15,<0.16`.

setup.py

@@ -100,7 +100,7 @@ def has_ext_modules(self):
         'pandas>=0.24,<1',
 
         # Dependency for mutual information computation.
-        'scikit-learn>=0.18,<0.21',
+        'scikit-learn>=0.18,<0.22',
 
         # TODO(pachristopher): Consider using multi-processing provided by
         # Beam's DirectRunner.