Update the setup.py to match the requirements.txt.

PiperOrigin-RevId: 471128177
tensorflow · Aug 31, 2022 · 628e5bb · 628e5bb
1 parent ed73077
commit 628e5bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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',