code: def parse_arguments():
      parser = argparse.ArgumentParser()
  
      parser.add_argument(
          "--log_dir",
          type=str,
          default=BENCH_LOG_DIR,
          help=f'Folder of performance sweep logs. Default is {os.path.join(".", BENCH_LOG_DIR)}',
      )
  
      args = parser.parse_args()
      print(f"args = {args}")
  
      return args
  
file_path: DeepSpeed/csrc/aio/py_test/aio_bench_generate_param.py
imports: import os
  import argparse
  import json
  from parse_aio_stats import READ_SPEED, WRITE_SPEED, get_sorted_results
  from perf_sweep_utils import BENCH_LOG_DIR, READ_LOG_DIR, WRITE_LOG_DIR
label_desc: Use image node_id 1 for calling a global function with example usage: parse_arguments() and returns: args