code: def arguments_collate_fn(
      parser: argparse.ArgumentParser,
  ) -> argparse.ArgumentParser:
      """Add arguments related to collate function"""
      group = parser.add_argument_group("Collate function arguments")
      group.add_argument(
          "--dataset.collate-fn-name-train",
          type=str,
          default="pytorch_default_collate_fn",
          help="Name of collate function for training. Defaults to pytorch_default_collate_fn.",
      )
      group.add_argument(
          "--dataset.collate-fn-name-val",
          type=str,
          default="pytorch_default_collate_fn",
          help="Name of collate function for validation. Defaults to pytorch_default_collate_fn.",
      )
      group.add_argument(
          "--dataset.collate-fn-name-test",
          type=str,
          default="pytorch_default_collate_fn",
          help="Name of collate function used for evaluation. "
          "Default is pytorch_default_collate_fn.",
      )
      return parser
  
imports: import argparse
  from utils import logger
  from utils.registry import Registry
label_desc: Use image node_id 1 for calling a global function with example usage: arguments_collate_fn(parser) and returns: parser