diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py')
| -rw-r--r-- | tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py b/tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py index 545471907f..eacdfd9ccc 100644 --- a/tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py +++ b/tensorflow/contrib/distributions/python/kernel_tests/conditional_transformed_distribution_test.py @@ -22,7 +22,7 @@ import numpy as np from tensorflow.contrib import distributions from tensorflow.contrib.distributions.python.kernel_tests import transformed_distribution_test -from tensorflow.contrib.distributions.python.ops.bijectors.conditional_bijector import ConditionalBijector +from tensorflow.contrib.distributions.python.ops.bijectors import conditional_bijector from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops @@ -32,7 +32,7 @@ from tensorflow.python.platform import test ds = distributions -class _ChooseLocation(ConditionalBijector): +class _ChooseLocation(conditional_bijector.ConditionalBijector): """A Bijector which chooses between one of two location parameters.""" def __init__(self, loc, name="ChooseLocation"): |