code: def _maybe_seed(s, sampler_type="stateful"):
      if (
          tf.executing_eagerly()
          and not JAX_MODE
          and sampler_type != "stateless"
      ):
          tf.random.set_seed(s)
          return None
      return s
  
imports: import collections
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import identity
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import hmc
  from tensorflow_probability.python.mcmc import sample_annealed_importance
  from tensorflow_probability.python.mcmc import transformed_kernel
label_desc: Use image node_id 1 for calling a global function with example usage: _maybe_seed(s, sampler_type) and returns: s, None