code: def _build_random_dataset(
      self, num_elements=10, seed=None, rerandomize_each_iteration=None
  ):
      dataset = dataset_ops.Dataset.random(
          seed=seed,
          rerandomize_each_iteration=rerandomize_each_iteration,
      )
      # Checkpoint tests need the test dataset to be finite whereas `random` is
      # infinite. Use `take` to limit the number of elements.
      return dataset.take(num_elements)
  
imports: import warnings
  from absl.testing import parameterized
  from tensorflow.python import tf2
  from tensorflow.python.data.kernel_tests import checkpoint_test_base
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import random_seed
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the RandomCheckpointTest obj's underlying member method code with example usage: obj._build_random_dataset(num_elements, seed, rerandomize_each_iteration) and returns: dataset