code: def main() -> None:
      parser = argparse.ArgumentParser(
          description="Generate ONNX diagnostics rules doc in markdown."
      )
      parser.add_argument(
          "out_dir",
          metavar="OUT_DIR",
          help="path to output directory for docs",
      )
      args = parser.parse_args()
      gen_docs(args.out_dir)
  
imports: import argparse
  import os
  from dataclasses import fields
  from torch.onnx._internal import diagnostics
  from torch.onnx._internal.diagnostics import infra
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types