forked from 626_privacy/tensorflow_privacy
Add direct sklearn
dependency.
PiperOrigin-RevId: 427925353
This commit is contained in:
parent
592ed995f1
commit
9d59eabd40
2 changed files with 2 additions and 0 deletions
|
@ -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
|
||||
|
|
1
setup.py
1
setup.py
|
@ -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',
|
||||
|
|
Loading…
Reference in a new issue