def enhance_data(raw_df):
  """Enhance the setup.tsv file with additional information necessary to build the annotation interfance.

  Args:
    raw_df: Dataframe representing the setup.tsv file (see readme)

  Returns:
    Dataframe that is used in Stage 2
  """

  df = pd.concat([
      raw_df,
      raw_df.apply(
          lambda row: wide_representation(row[0]), result_type='expand', axis=1)
  ],
                 axis=1)

  return df
Enhance the setup.tsv file with additional information necessary to build the annotation interfance.

  Args:
    raw_df: Dataframe representing the setup.tsv file (see readme)

  Returns:
    Dataframe that is used in Stage 2
  
language/asqa/human_annotation/preparation.py