code: def process_data(
      train_data, val_data, save_name_base, include_topic, sft_format
  ):
      train_set = extract_questions(
          train_data, include_topic, sft_format, split="train"
      )
      val_set = extract_questions(
          val_data, include_topic, sft_format, split="val"
      )
      test_set = extract_questions(
          val_data, include_topic, sft_format, split="test"
      )
  
      gen_file(train_set, save_name_base, "train", sft_format)
      gen_file(val_set, save_name_base, "val", sft_format)
      gen_file(
          test_set,
          save_name_base,
          "test",
          sft_format,
          make_ground_truth=True,
      )
      gen_file(
          test_set,
          save_name_base,
          "test",
          sft_format,
          make_ground_truth=False,
      )
  
imports: import argparse
  import json
  from tqdm import tqdm
label_desc: Use image node_id 2 for calling a global function with example usage: process_data(train_data, val_data, save_name_base, include_topic, sft_format) without return types