Add dependency on tensorflow probability.

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

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

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