code: def testSqrtWithFiniteGradsHasCorrectValues(self):
      self.assertTrue(
          np.isnan(self.evaluate(util.sqrt_with_finite_grads(-1.0)))
      )
      xs = np.linspace(0.0, 10.0, 100)
      self.assertAllEqual(
          self.evaluate(tf.sqrt(xs)),
          self.evaluate(util.sqrt_with_finite_grads(xs)),
      )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.gradient import value_and_gradient
  from tensorflow_probability.python.math.psd_kernels.internal import util
label_desc: Use image node_id 8 for calling the UtilTest obj's underlying member method code with example usage: obj.testSqrtWithFiniteGradsHasCorrectValues() without return types