code: def load_squad_dataset(with_metadata=False):
      dataset = datasets.load_dataset("squad", split="train")
      answers = []
      questions = []
      context = []
      title = []
      for row in dataset:
          context.append(row[CONTEXT])
          questions.append(row[QUESTION])
          answers.append(row[ANSWERS]["text"][0])
          if with_metadata:
              title.append(row[TITLE])
      columns = {
          CONTEXT: context,
          QUESTIONS: questions,
          ANSWERS: answers,
      }
      if with_metadata:
          columns.update({TITLE: title})
      data = pd.DataFrame(columns)
      return data
  
imports: import os
  import zipfile
  import datasets
  import pandas
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
  from raiutils.common.retries import retry_function
  from responsibleai_text.common.constants import QuestionAnsweringFields
label_desc: Use image node_id 2 for calling a global function with example usage: load_squad_dataset(with_metadata) and returns: data