From 628e5bb926ef5607e9762b3e85eec1c5be235c25 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Tue, 30 Aug 2022 17:16:14 -0700
Subject: [PATCH] Update the `setup.py` to match the `requirements.txt`.

PiperOrigin-RevId: 471128177
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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',