code: def testTurnOnStoreParametersInKernelResults(self):
      mvn = mvn_diag.MultivariateNormalDiag(
          self.mvn_mean, scale_diag=self.mvn_scale
      )
      target_log_prob_fn = mvn.log_prob
      kernel = phmc.PreconditionedHamiltonianMonteCarlo(
          target_log_prob_fn=target_log_prob_fn,
          num_leapfrog_steps=2,
          step_size=1.0,
      )
      self.assertFalse(kernel.parameters["store_parameters_in_results"])
      initial_running_variance = (
          sample_stats.RunningVariance.from_stats(
              num_samples=1.0, mean=tf.zeros(3), variance=tf.ones(3)
          )
      )
      kernel = dmma.DiagonalMassMatrixAdaptation(
          inner_kernel=kernel,
          initial_running_variance=initial_running_variance,
      )
      self.assertTrue(
          kernel.inner_kernel.parameters["store_parameters_in_results"]
      )
  
imports: import collections
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import joint_distribution_sequential
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import sample
  from tensorflow_probability.python.experimental.mcmc import diagonal_mass_matrix_adaptation
  from tensorflow_probability.python.experimental.mcmc import preconditioned_hmc
  from tensorflow_probability.python.experimental.mcmc import preconditioning_utils
  from tensorflow_probability.python.experimental.stats import sample_stats
  from tensorflow_probability.python.internal import distribute_lib
  from tensorflow_probability.python.internal import distribute_test_lib
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import sample
label_desc: Use image node_id 2 for calling the DiagonalMassMatrixAdaptationTest obj's underlying member method code with example usage: obj.testTurnOnStoreParametersInKernelResults() without return types