diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py')
| -rw-r--r-- | tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py b/tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py index af675787b1..7fee2e1f3a 100644 --- a/tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py +++ b/tensorflow/contrib/distributions/python/ops/relaxed_bernoulli.py @@ -21,9 +21,7 @@ from __future__ import print_function from tensorflow.contrib.distributions.python.ops import distribution_util from tensorflow.contrib.distributions.python.ops import logistic from tensorflow.contrib.distributions.python.ops import transformed_distribution -# Bijectors must be directly imported because `remove_undocumented` prevents -# individual file imports. -from tensorflow.contrib.distributions.python.ops.bijectors.sigmoid import Sigmoid +from tensorflow.contrib.distributions.python.ops.bijectors import sigmoid as sigmoid_lib from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops @@ -178,7 +176,7 @@ class RelaxedBernoulli(transformed_distribution.TransformedDistribution): validate_args=validate_args, allow_nan_stats=allow_nan_stats, name=name + "/Logistic"), - bijector=Sigmoid(validate_args=validate_args), + bijector=sigmoid_lib.Sigmoid(validate_args=validate_args), validate_args=validate_args, name=name) self._parameters = parameters |