code: def setUp(self):
      super().setUp()
  
      if JAX_MODE and FLAGS.test_tfp_jax_prng != "default":
          from jax import config  # pylint: disable=g-import-not-at-top
  
          config.update(
              "jax_default_prng_impl", FLAGS.test_tfp_jax_prng
          )
  
file_path: probability/tensorflow_probability/python/internal/samplers_test.py
imports: from absl import flags
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the RandomTest obj's underlying member method code with example usage: obj.setUp() without return types