diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py')
-rw-r--r-- | tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py index 267e4ad350..7b09553be9 100644 --- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py +++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/cholesky_outer_product_test.py @@ -21,7 +21,7 @@ from __future__ import print_function from tensorflow.contrib.distributions.python.ops import bijectors from tensorflow.contrib.distributions.python.ops import gamma as gamma_lib from tensorflow.contrib.distributions.python.ops import transformed_distribution as transformed_distribution_lib -from tensorflow.contrib.distributions.python.ops.bijectors.bijector_test_util import assert_scalar_congruency +from tensorflow.contrib.distributions.python.ops.bijectors import bijector_test_util from tensorflow.python.framework import tensor_shape from tensorflow.python.ops import array_ops from tensorflow.python.platform import test @@ -57,7 +57,7 @@ class InvertBijectorTest(test.TestCase): def testScalarCongruency(self): with self.test_session(): bijector = bijectors.Invert(bijectors.Exp()) - assert_scalar_congruency( + bijector_test_util.assert_scalar_congruency( bijector, lower_x=1e-3, upper_x=1.5, rtol=0.05) def testShapeGetters(self): |