1 files changed, 11 insertions, 10 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_test.py
index 9a96b91c6d..e16f9dff22 100644
--- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_test.py
+++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/sigmoid_test.py
@@ -21,9 +21,8 @@ from __future__ import print_function
 import numpy as np
 from scipy import special
 
-from tensorflow.contrib.distributions.python.ops.bijectors.bijector_test_util import assert_bijective_and_finite
-from tensorflow.contrib.distributions.python.ops.bijectors.bijector_test_util import assert_scalar_congruency
-from tensorflow.contrib.distributions.python.ops.bijectors.sigmoid import Sigmoid
+from tensorflow.contrib.distributions.python.ops.bijectors import bijector_test_util
+from tensorflow.contrib.distributions.python.ops.bijectors import sigmoid
 from tensorflow.python.platform import test
 
 
@@ -32,33 +31,35 @@ class SigmoidBijectorTest(test.TestCase):
 
   def testBijector(self):
     with self.test_session():
-      self.assertEqual("sigmoid", Sigmoid().name)
+      self.assertEqual("sigmoid", sigmoid.Sigmoid().name)
       x = np.linspace(-10., 10., 100).reshape([2, 5, 10]).astype(np.float32)
       y = special.expit(x)
       ildj = -np.log(y) - np.log1p(-y)
       self.assertAllClose(
-          y, Sigmoid().forward(x).eval(),
+          y, sigmoid.Sigmoid().forward(x).eval(),
           atol=0., rtol=1e-2)
       self.assertAllClose(
-          x, Sigmoid().inverse(y).eval(),
+          x, sigmoid.Sigmoid().inverse(y).eval(),
           atol=0., rtol=1e-4)
       self.assertAllClose(
-          ildj, Sigmoid().inverse_log_det_jacobian(y).eval(),
+          ildj, sigmoid.Sigmoid().inverse_log_det_jacobian(y).eval(),
           atol=0., rtol=1e-6)
       self.assertAllClose(
-          -ildj, Sigmoid().forward_log_det_jacobian(x).eval(),
+          -ildj, sigmoid.Sigmoid().forward_log_det_jacobian(x).eval(),
           atol=0., rtol=1e-4)
 
   def testScalarCongruency(self):
     with self.test_session():
-      assert_scalar_congruency(Sigmoid(), lower_x=-7., upper_x=7.)
+      bijector_test_util.assert_scalar_congruency(
+          sigmoid.Sigmoid(), lower_x=-7., upper_x=7.)
 
   def testBijectiveAndFinite(self):
     with self.test_session():
       x = np.linspace(-7., 7., 100).astype(np.float32)
       eps = 1e-3
       y = np.linspace(eps, 1. - eps, 100).astype(np.float32)
-      assert_bijective_and_finite(Sigmoid(), x, y, atol=0., rtol=1e-4)
+      bijector_test_util.assert_bijective_and_finite(
+          sigmoid.Sigmoid(), x, y, atol=0., rtol=1e-4)
 
 
 if __name__ == "__main__":