code: def main():
      parser = argparse.ArgumentParser()
      parser.add_argument("--data-dir", type=str, default=".")
      parser.add_argument(
          "--train-file", type=str, default="train-v1.1.json"
      )
      parser.add_argument(
          "--dev-file", type=str, default="dev-v1.1.json"
      )
      parser.add_argument("--save-name-base", type=str, default="squad")
      parser.add_argument("--include-topic-name", action="store_true")
      parser.add_argument("--random-seed", type=int, default=1234)
      parser.add_argument("--sft-format", action="store_true")
      args = parser.parse_args()
  
      train_data_dict = json.load(
          open(f"{args.data_dir}/{args.train_file}")
      )
      dev_data_dict = json.load(
          open(f"{args.data_dir}/{args.dev_file}")
      )
      train_data = train_data_dict["data"]
      val_data = dev_data_dict["data"]
  
      save_name_base = f"{args.data_dir}/{args.save_name_base}"
  
      process_data(
          train_data,
          val_data,
          save_name_base,
          args.include_topic_name,
          args.sft_format,
      )
  
imports: import argparse
  import json
  from tqdm import tqdm
label_desc: Use image node_id 1 for calling a global function with example usage: main() without return types