code: def testDocstring2(self):
      Root = (
          jdc.JointDistributionCoroutine.Root
      )  # pylint: disable=invalid-name
  
      def model():
          e = yield Root(
              independent.Independent(
                  exponential.Exponential(rate=[100, 120]), 1
              )
          )
          g = yield gamma.Gamma(concentration=e[..., 0], rate=e[..., 1])
          n = yield Root(normal.Normal(loc=0, scale=2.0))
          yield normal.Normal(loc=n, scale=g)
  
      joint = jdc.JointDistributionCoroutine(model)
      d = blockwise.Blockwise(joint, validate_args=True)
  
      x = d.sample([2, 1], seed=test_util.test_seed())
      y = d.log_prob(x)
      x_, y_ = self.evaluate([x, y])
      self.assertEqual((2, 1, 2 + 1 + 1 + 1), x_.shape)
      self.assertDTypeEqual(x, np.float32)
      self.assertEqual((2, 1), y_.shape)
      self.assertDTypeEqual(y, np.float32)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import blockwise
  from tensorflow_probability.python.distributions import exponential
  from tensorflow_probability.python.distributions import gamma
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import joint_distribution_coroutine
  from tensorflow_probability.python.distributions import joint_distribution_named
  from tensorflow_probability.python.distributions import joint_distribution_sequential
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import mvn_linear_operator
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the BlockwiseTest obj's underlying member method code with example usage: obj.testDocstring2() without return types