From c8a49e1039b7720a6c9cc919b44311e7737b70bd Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Tue, 15 Feb 2022 18:08:22 +0000
Subject: [PATCH] Update the direct TensorFlow dependency to `~=2.4`.

This means that we would accept a hot fix and a new minor version , but not accept major version changes without releasing TFP.

PiperOrigin-RevId: 428811702
---
 requirements.txt | 4 ++--
 setup.py         | 9 ++-------
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index d6cfb1f..cc3aafd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,6 +8,6 @@ pandas~=1.1.4
 scipy~=1.2.1
 sklearn~=1.0.2
 tensorflow-datasets~=4.5.2
-tensorflow-estimator>=2.3.0
+tensorflow-estimator~=2.4
 tensorflow-probability~=0.15.0
-tensorflow>=1.14
+tensorflow~=2.4
diff --git a/setup.py b/setup.py
index 98a6f97..842c141 100644
--- a/setup.py
+++ b/setup.py
@@ -36,13 +36,8 @@ setup(
         'scipy~=1.2.1',
         'sklearn~=1.0.2',
         'tensorflow-datasets~=4.5.2',
-        'tensorflow-estimator>=2.3.0',
+        'tensorflow-estimator~=2.4',
         'tensorflow-probability~=0.15.0',
+        'tensorflow~=2.4',
     ],
-    # Explicit dependence on TensorFlow is not supported.
-    # See https://github.com/tensorflow/tensorflow/issues/7166
-    extras_require={
-        'tf': ['tensorflow>=1.0.0'],
-        'tf_gpu': ['tensorflow-gpu>=1.0.0'],
-    },
     packages=find_packages())