code: def _get_dataset_size(self, files):
      if tf.executing_eagerly():
          return sum(
              1
              for _ in tf.data.TFRecordDataset(
                  files, compression_type="GZIP"
              )
          )
      else:
          result = 0
          for file in files:
              result += sum(
                  1
                  for _ in tf.compat.v1.io.tf_record_iterator(
                      file,
                      tf.io.TFRecordOptions(compression_type="GZIP"),
                  )
              )
          return result
  
imports: import copy
  import json
  import os
  import re
  import tempfile
  from unittest import mock
  from absl.testing import parameterized
  import apache_beam
  import tensorflow
  import tensorflow_data_validation
  import tensorflow_transform
  from tensorflow_transform.beam import tft_unit
  from tfx import types
  from tfx.components.testdata.module_file import transform_module
  from tfx.components.transform import executor
  from tfx.dsl.io import fileio
  from tfx.proto import example_gen_pb2
  from tfx.proto import transform_pb2
  from tfx.types import artifact_utils
  from tfx.types import standard_artifacts
  from tfx.types import standard_component_specs
  from tfx.utils import io_utils
  from tfx.utils import name_utils
  from tfx.utils import proto_utils
label_desc: Use image node_id 2 for calling the ExecutorTest obj's underlying member method code with example usage: obj._get_dataset_size(files) and returns: sum, result