626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Add dependency on tensorflow probability.

Browse source 
PiperOrigin-RevId: 389962756
This commit is contained in:
Galen Andrew 2021-08-10 13:23:10 -07:00 committed by A. Unique TensorFlower
parent dad4ff0a58
commit fbaa55cc43
2 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
requirements.txt
View file

@ -4,3 +4,4 @@ tensorflow-estimator>=2.3.0
attrs>=21.2.0
mpmath
dm-tree~=0.1.1
tensorflow-probability>=0.13.0

1
setup.py
View file

@ -26,6 +26,7 @@ setup(
'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.
],
# Explicit dependence on TensorFlow is not supported.
# See https://github.com/tensorflow/tensorflow/issues/7166