Update the Python dependency for sklearn package.

PiperOrigin-RevId: 428896276
2022-02-15 23:54:05 +00:00 · 2022-02-15 23:54:05 +00:00 · 8544af2500
commit 8544af2500
parent 11b4c318a2
2 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -6,7 +6,7 @@ mpmath
 numpy~=1.21.5
 pandas~=1.1.4
 scipy~=1.2.1
-sklearn~=1.0.2
+scikit-learn~=1.0.2
 tensorflow-datasets~=4.5.2
 tensorflow-estimator~=2.4
 tensorflow-probability~=0.15.0
--- a/setup.py
+++ b/setup.py
@ -34,7 +34,7 @@ setup(
        'numpy~=1.21.5',
        'pandas~=1.1.4',
        'scipy~=1.2.1',
-        'sklearn~=1.0.2',
+        'scikit-learn~=1.0.2',
        'tensorflow-datasets~=4.5.2',
        'tensorflow-estimator~=2.4',
        'tensorflow-probability~=0.15.0',