diff --git a/requirements.txt b/requirements.txt index 304b0cd..30e1932 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,8 @@ -tensorflow>=1.14 -scipy>=0.17 -tensorflow-estimator>=2.3.0 attrs>=21.2.0 -mpmath dm-tree~=0.1.1 -tensorflow-probability>=0.13.0 +mpmath +scipy>=0.17 tensorflow-datasets>=4.4.0 +tensorflow-estimator>=2.3.0 +tensorflow-probability>=0.13.0 +tensorflow>=1.14 diff --git a/setup.py b/setup.py index 9d1d596..8f86d33 100644 --- a/setup.py +++ b/setup.py @@ -21,13 +21,13 @@ setup( url='https://github.com/tensorflow/privacy', license='Apache-2.0', install_requires=[ - '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 + 'mpmath', # used in tests only + 'scipy>=0.17', + 'tensorflow-datasets>=4.4.0', + 'tensorflow-estimator>=2.3.0', # for DP versions of estimator. '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