code: def testUniformNans(self):
      a = 10.0
      b = [11.0, 100.0]
      dist = uniform.Uniform(low=a, high=b, validate_args=False)
  
      no_nans = tf.constant(1.0)
      nans = tf.constant(0.0) / tf.constant(0.0)
      self.assertTrue(self.evaluate(tf.math.is_nan(nans)))
      with_nans = tf.stack([no_nans, nans])
  
      pdf = dist.prob(with_nans)
  
      is_nan = self.evaluate(tf.math.is_nan(pdf))
      self.assertFalse(is_nan[0])
      self.assertTrue(is_nan[1])
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 14 for calling the UniformTest obj's underlying member method code with example usage: obj.testUniformNans() without return types