code: def test_sanitize_tensor_or_tensorlike(self):
      # TODO(b/223267515): In JAX mode, test that we raise a user-friendly error
      # when the seed is not a PRNGKey.
      seed1 = samplers.sanitize_seed([0, 1])
      seed2 = samplers.sanitize_seed(np.array([0, 1]))
      seed1, seed2 = self.evaluate([seed1, seed2])
      self.assertAllEqual(seed1, seed2)
  
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 6 for calling the RandomTest obj's underlying member method code with example usage: obj.test_sanitize_tensor_or_tensorlike() without return types