Add direct sklearn dependency.

PiperOrigin-RevId: 427925353
This commit is contained in:
Michael Reneer 2022-02-10 22:32:36 -08:00 committed by A. Unique TensorFlower
parent 592ed995f1
commit 9d59eabd40
2 changed files with 2 additions and 0 deletions

View file

@ -6,6 +6,7 @@ mpmath
numpy~=1.21.5
pandas~=1.1.4
scipy~=1.2.1
sklearn~=1.0.2
tensorflow-datasets>=4.4.0
tensorflow-estimator>=2.3.0
tensorflow-probability>=0.13.0

View file

@ -34,6 +34,7 @@ setup(
'numpy~=1.21.5',
'pandas~=1.1.4',
'scipy~=1.2.1',
'sklearn~=1.0.2',
'tensorflow-datasets>=4.4.0',
'tensorflow-estimator>=2.3.0',
'tensorflow-probability>=0.13.0',