diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py')
| -rw-r--r-- | tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py index 26e0d2a539..9c2a1c8f91 100644 --- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py +++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/conditional_bijector_test.py @@ -18,12 +18,12 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -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.platform import test -class _TestBijector(ConditionalBijector): +class _TestBijector(conditional_bijector.ConditionalBijector): def __init__(self): super(_TestBijector, self).__init__( |