code: def test_get_likelihood_model(self, likelihood_model):
      m = likelihoods.get_likelihood_model(likelihood_model, {})
      lp = m.log_likelihood(
          params={"noise_scale": 0.4},
          nn_out=jnp.ones(shape=(10, m.num_outputs())),
          observations=jnp.zeros(shape=(10, 1)),
      )
      self.assertEqual(lp.shape, (10, 1))
  
      m2 = likelihoods.get_likelihood_model(
          likelihood_model,
          {
              "noise_min": 0.1,
              "log_noise_scale": 0.5,
              "log_noise_mean": -1.0,
          },
      )
      lp2 = m2.log_likelihood(
          params={"noise_scale": 0.4},
          nn_out=jnp.ones(shape=(10, m2.num_outputs())),
          observations=jnp.zeros(shape=(10, 1)),
      )
      self.assertEqual(lp2.shape, (10, 1))
  
imports: from absl.testing import parameterized
  import jax.numpy
  from tensorflow_probability.python.experimental.autobnn import likelihoods
  from absl.testing import absltest
label_desc: Use image node_id 2 for calling the LikelihoodTests obj's underlying member method code with example usage: obj.test_get_likelihood_model(likelihood_model) without return types