code: def _build_dataset(self, num_elements, upper_bound, options=None):
      dataset = dataset_ops.Dataset.range(num_elements)
      dataset = dataset.take_while(predicate=lambda x: x < upper_bound)
      if options:
          dataset = dataset.with_options(options)
      return dataset
  
imports: from absl.testing import parameterized
  import numpy
  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.framework import combinations
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import errors
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the TakeWhileCheckpointTest obj's underlying member method code with example usage: obj._build_dataset(num_elements, upper_bound, options) and returns: dataset