code: def _build_shuffle_dataset(
      self,
      range_limit=10,
      num_repeats=5,
      buffer_size=5,
      seed=None,
      reshuffle_each_iteration=None,
      symbolic_checkpoint=None,
  ):
      dataset = (
          dataset_ops.Dataset.range(range_limit)
          .shuffle(
              buffer_size,
              seed=seed,
              reshuffle_each_iteration=reshuffle_each_iteration,
          )
          .repeat(num_repeats)
      )
  
      if symbolic_checkpoint:
          options = options_lib.Options()
          options.experimental_symbolic_checkpoint = symbolic_checkpoint
          dataset = dataset.with_options(options)
  
      return dataset
  
imports: import collections
  import functools
  import sys
  from absl.testing import parameterized
  import numpy
  from tensorflow.python import pywrap_sanitizers
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.checkpoint import checkpoint_management
  from tensorflow.python.data.experimental.ops import iterator_ops
  from tensorflow.python.data.experimental.ops import random_access
  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.data.ops import options
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import random_seed
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import check_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.training import saver
label_desc: Use image node_id 1 for calling the ShuffleCheckpointTest obj's underlying member method code with example usage: obj._build_shuffle_dataset(range_limit, num_repeats, buffer_size, seed, reshuffle_each_iteration, symbolic_checkpoint) and returns: dataset