diff --git a/setup.py b/setup.py index c39e54b0..7bf038f5 100644 --- a/setup.py +++ b/setup.py @@ -27,14 +27,14 @@ url='https://github.com/tensorflow/privacy', license='Apache-2.0', install_requires=[ - 'absl-py~=1.0', + 'absl-py>=1.0,==1.*', 'attrs~=21.4', 'dm-tree==0.1.7', 'dp-accounting==0.3.0', 'matplotlib~=3.3', 'numpy~=1.21', 'pandas~=1.4', - 'scikit-learn~=1.0', + 'scikit-learn>=1.0,==1.*', 'scipy~=1.7', 'tensorflow-datasets~=4.5', 'tensorflow-estimator~=2.4',