From f44dcb8760af62879cae2d2b5b102ccf524b83d2 Mon Sep 17 00:00:00 2001
From: Galen Andrew <galenandrew@google.com>
Date: Wed, 11 Aug 2021 10:53:59 -0700
Subject: [PATCH] Add tensorflow and tensorflow-datasets to setup/requirements.

PiperOrigin-RevId: 390171562
---
 requirements.txt              | 1 +
 setup.py                      | 4 +++-
 tensorflow_privacy/version.py | 2 +-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 7994d55..304b0cd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ attrs>=21.2.0
 mpmath
 dm-tree~=0.1.1
 tensorflow-probability>=0.13.0
+tensorflow-datasets>=4.4.0
diff --git a/setup.py b/setup.py
index 4ca4064..9a698d1 100644
--- a/setup.py
+++ b/setup.py
@@ -17,16 +17,18 @@ from setuptools import setup
 
 setup(
     name='tensorflow_privacy',
-    version='0.7.0',
+    version='0.7.1',
     url='https://github.com/tensorflow/privacy',
     license='Apache-2.0',
     install_requires=[
+        'tensorflow>=1.14',
         'scipy>=0.17',
         'tensorflow-estimator>=2.3.0',  # for DP versions of estimator.
         'attrs>=21.2.0',  # for tree_aggregation_query.py.
         'mpmath',  # used in tests only
         'dm-tree~=0.1.1',  # used in tests only
         'tensorflow-probability>=0.13.0',  # For discrete Gaussian.
+        'tensorflow-datasets>=4.4.0'
     ],
     # Explicit dependence on TensorFlow is not supported.
     # See https://github.com/tensorflow/tensorflow/issues/7166
diff --git a/tensorflow_privacy/version.py b/tensorflow_privacy/version.py
index da4254b..878eae5 100644
--- a/tensorflow_privacy/version.py
+++ b/tensorflow_privacy/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 """TensorFlow Privacy version."""
 
-__version__ = '0.7.0'
+__version__ = '0.7.1'