code: def testSamplesAtBoundaryNotNaN(self):
      temperature = 1e-2
      dist = relaxed_bernoulli.RelaxedBernoulli(
          temperature, probs=1.0, validate_args=True
      )
      self.assertFalse(
          np.any(
              np.isnan(
                  self.evaluate(
                      dist.log_prob(
                          dist.sample(10, seed=test_util.test_seed())
                      )
                  )
              )
          )
      )
  
      dist = relaxed_bernoulli.RelaxedBernoulli(
          temperature, probs=0.0, validate_args=True
      )
      self.assertFalse(
          np.any(
              np.isnan(
                  self.evaluate(
                      dist.log_prob(
                          dist.sample(10, seed=test_util.test_seed())
                      )
                  )
              )
          )
      )
  
imports: import numpy
  import scipy.special
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import relaxed_bernoulli
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 9 for calling the RelaxedBernoulliTest obj's underlying member method code with example usage: obj.testSamplesAtBoundaryNotNaN() without return types