code: def __init__(
      self,
      input_config,
      output_config=None,
      range_config=None,
      output_data_format=example_gen_pb2.FORMAT_TF_EXAMPLE,
      output_file_format=example_gen_pb2.FORMAT_TFRECORDS_GZIP,
  ):
      super().__init__(
          input_config=input_config,
          output_config=output_config,
          range_config=range_config,
          output_data_format=output_data_format,
          output_file_format=output_file_format,
      )
  
imports: import tensorflow
  from tfx.components.example_gen import base_example_gen_executor
  from tfx.components.example_gen import component
  from tfx.components.example_gen import driver
  from tfx.dsl.components.base import executor_spec
  from tfx.proto import example_gen_pb2
  from tfx.proto import range_config_pb2
  from tfx.types import standard_artifacts
  from tfx.types import standard_component_specs
  from tfx.utils import proto_utils
  from google.protobuf import any_pb2
label_desc: Use image node_id 1 to create a new TestQueryBasedExampleGenComponent object from inherited base classes: component with example: obj = TestQueryBasedExampleGenComponent(input_config, output_config, range_config, output_data_format, output_file_format)