From 669b72fa8657f12d8b7529b34d52366eef714165 Mon Sep 17 00:00:00 2001 From: Brian Patton Date: Fri, 12 Jul 2019 13:04:10 -0700 Subject: [PATCH] Suppresses the tf_should_use warnings in test logs. PiperOrigin-RevId: 257854575 --- tensorflow_probability/python/bijectors/bijector.py | 3 +++ tensorflow_probability/python/distributions/distribution.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/tensorflow_probability/python/bijectors/bijector.py b/tensorflow_probability/python/bijectors/bijector.py index 0380f28748..e0899d3cbb 100644 --- a/tensorflow_probability/python/bijectors/bijector.py +++ b/tensorflow_probability/python/bijectors/bijector.py @@ -636,6 +636,9 @@ def __init__(self, self._initial_parameter_control_dependencies = tuple( d for d in self._parameter_control_dependencies(is_init=True) if d is not None) + if self._initial_parameter_control_dependencies: + self._initial_parameter_control_dependencies = ( + tf.group(*self._initial_parameter_control_dependencies),) if forward_min_event_ndims is None and inverse_min_event_ndims is None: raise ValueError('Must specify at least one of `forward_min_event_ndims` ' diff --git a/tensorflow_probability/python/distributions/distribution.py b/tensorflow_probability/python/distributions/distribution.py index 5ae02fc14d..9045b03179 100644 --- a/tensorflow_probability/python/distributions/distribution.py +++ b/tensorflow_probability/python/distributions/distribution.py @@ -492,6 +492,9 @@ def __init__(self, self._initial_parameter_control_dependencies = tuple( d for d in self._parameter_control_dependencies(is_init=True) if d is not None) + if self._initial_parameter_control_dependencies: + self._initial_parameter_control_dependencies = ( + tf.group(*self._initial_parameter_control_dependencies),) @classmethod def param_shapes(cls, sample_shape, name='DistributionParamShapes'):