code: def ExtractSliceKeys(
      extracts: beam.pvalue.PCollection,
      slice_spec: List[slicer.SingleSliceSpec],
      eval_config: Optional[config_pb2.EvalConfig] = None,
      materialize: bool = True,
  ) -> beam.pvalue.PCollection:
      return extracts | "ExtractSliceKeys" >> beam.ParDo(
          ExtractSliceKeysFn(eval_config, materialize),
          slice_spec=slice_spec,
      )
  
file_path: model-analysis/tensorflow_model_analysis/extractors/slice_key_extractor.py
imports: import copy
  from typing import List, Optional
  import apache_beam
  from tensorflow_model_analysis import constants
  from tensorflow_model_analysis.api import types
  from tensorflow_model_analysis.extractors import extractor
  from tensorflow_model_analysis.proto import config_pb2
  from tensorflow_model_analysis.slicer import slicer_lib
  from tensorflow_model_analysis.utils import util
label_desc: Use image node_id 2 for calling a global function with example usage: ExtractSliceKeys(extracts, slice_spec, eval_config, materialize) and returns: unknown