Skip to content

Commit

Permalink
Suppresses the tf_should_use warnings in test logs.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 257854575
  • Loading branch information
brianwa84 authored and tensorflower-gardener committed Jul 12, 2019
1 parent 185c15f commit 669b72f
Showing 2 changed files with 6 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tensorflow_probability/python/bijectors/bijector.py
Original file line number Diff line number Diff line change
@@ -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` '
3 changes: 3 additions & 0 deletions tensorflow_probability/python/distributions/distribution.py
Original file line number Diff line number Diff line change
@@ -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'):

0 comments on commit 669b72f

Please sign in to comment.