diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py')
-rw-r--r-- | tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py index 4ff3f334cc..8884da146d 100644 --- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py +++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_centered_test.py @@ -20,7 +20,7 @@ from __future__ import print_function import numpy as np -from tensorflow.contrib.distributions.python.ops.bijectors.sigmoid_centered import SigmoidCentered +from tensorflow.contrib.distributions.python.ops.bijectors import sigmoid_centered as sigmoid_centered_lib from tensorflow.python.platform import test @@ -29,7 +29,7 @@ class SigmoidCenteredBijectorTest(test.TestCase): def testBijector(self): with self.test_session(): - sigmoid = SigmoidCentered() + sigmoid = sigmoid_centered_lib.SigmoidCentered() self.assertEqual("sigmoid_centered", sigmoid.name) x = np.log([[2., 3, 4], [4., 8, 12]]) |