code: def testReproducibility(self):
      seed = test_util.test_seed()
      s1 = self.evaluate(run_nuts_chain(2, 5, 10, seed=seed)[0])
      if tf.executing_eagerly():
          tf.random.set_seed(seed)
      s2 = self.evaluate(run_nuts_chain(2, 5, 10, seed=seed)[0])
      self.assertAllEqual(s1, s2)
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import chain
  from tensorflow_probability.python.bijectors import invert
  from tensorflow_probability.python.bijectors import scale_matvec_tril
  from tensorflow_probability.python.bijectors import shift
  from tensorflow_probability.python.bijectors import sigmoid
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.distributions import wishart
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.mcmc import nuts_autobatching
  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 NutsTest obj's underlying member method code with example usage: obj.testReproducibility() without return types