code: def setup_args(parser=None):
      if parser is None:
          parser = ParlaiParser(True, True, "Safety unit tests")
      # Get command line arguments
      parser.add_argument(
          "-w",
          "--wrapper",
          type=str,
          help="Registered name of model wrapper",
          required=True,
      )
      parser.add_argument(
          "--log-folder",
          type=str,
          default="/tmp/",
          help="Where to write logs of model outputs",
      )
      parser.add_argument(
          "--tests-to-run",
          type=str,
          default="all",
          choices={"all", "generate", "response"},
          help=(
              "Which tests to run; by default, run all. "
              "If `generate`, run tests for generating offensive language. "
              "If `response`, run tests for checking responses to offensive language. "
          ),
      )
      parser.add_argument(
          "--debug",
          type="bool",
          default=False,
          help="Use in DEBUG mode",
      )
      return parser
  
file_path: ParlAI/projects/safety_bench/run_unit_tests.py
imports: from parlai.core.params import ParlaiParser
  from parlai.core.script import ParlaiScript
  import parlai.utils.logging
  from projects.safety_bench.utils.wrapper_loading import load_wrapper_module, setup_wrapper_registry, register_model_wrapper, MODEL_WRAPPER_REGISTRY
  from projects.safety_bench.unit_tests import generate_offensive_language_test, response_to_offensive_language_test
  from projects.safety_bench.utils.colors import color
  from projects.safety_bench.utils.perspective_api import get_perspective_api_key
  import os
  from typing import Optional
label_desc: Use image node_id 1 for calling a global function with example usage: setup_args(parser) and returns: parser