code: def _fetch_raw_data_column(
      record_batch: pa.RecordBatch,
  ) -> np.ndarray:
      """Fetch the raw data column.
  
      Args:
        record_batch: An Arrow RecordBatch.
  
      Returns:
        Raw data column.
      """
      column_index = record_batch.schema.get_field_index(
          constants.ARROW_INPUT_COLUMN
      )
      assert column_index >= 0, "Arrow input column not found."
      return np.asarray(record_batch.column(column_index).flatten())
  
imports: import collections
  import copy
  from typing import Any, Dict, List, Optional, Sequence
  import apache_beam
  import numpy
  import pyarrow
  from tensorflow_model_analysis import constants
  from tensorflow_model_analysis.api import types
  from tensorflow_model_analysis.eval_saved_model import constants
  from tensorflow_model_analysis.extractors import extractor
  from tensorflow_model_analysis.extractors import legacy_feature_extractor
  from tensorflow_model_analysis.proto import config_pb2
  from tensorflow_model_analysis.utils import model_util
label_desc: Use image node_id 3 for calling a global function with example usage: _fetch_raw_data_column(record_batch) and returns: np