Add tensorflow and tensorflow-datasets to setup/requirements.

PiperOrigin-RevId: 390171562
This commit is contained in:
Galen Andrew 2021-08-11 10:53:59 -07:00 committed by A. Unique TensorFlower
parent fbaa55cc43
commit f44dcb8760
3 changed files with 5 additions and 2 deletions

View file

@ -5,3 +5,4 @@ attrs>=21.2.0
mpmath
dm-tree~=0.1.1
tensorflow-probability>=0.13.0
tensorflow-datasets>=4.4.0

View file

@ -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

View file

@ -13,4 +13,4 @@
# limitations under the License.
"""TensorFlow Privacy version."""
__version__ = '0.7.0'
__version__ = '0.7.1'