code: def test_get_params_batch_length(self):
      self.assertEqual(
          10,
          training_util.get_params_batch_length(
              {"noise_scale": jnp.ones(10)}
          ),
      )
      self.assertEqual(
          5,
          training_util.get_params_batch_length(
              {"noise_scale": jnp.ones((5, 7))}
          ),
      )
      self.assertEqual(
          20,
          training_util.get_params_batch_length(
              {"params": {"noise_scale": jnp.ones((20, 8))}}
          ),
      )
  
imports: import chex
  import jax
  import jax.numpy
  import numpy
  from tensorflow_probability.python.experimental.autobnn import kernels
  from tensorflow_probability.python.experimental.autobnn import operators
  from tensorflow_probability.python.experimental.autobnn import training_util
  from tensorflow_probability.python.experimental.autobnn import util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 5 for calling the TrainingUtilTest obj's underlying member method code with example usage: obj.test_get_params_batch_length() without return types