code: def parse_args():
      parser = argparse.ArgumentParser()
  
      parser.add_argument(
          "--data-directory",
          help=(
              "Name of the S3 directory in the air-example-data-2 "
              "bucket to load data from."
          ),
      )
      parser.add_argument(
          "--data-format",
          choices=["parquet", "raw"],
          help="The format of the data. Can be either parquet or raw.",
      )
      parser.add_argument(
          "--smoke-test",
          action="store_true",
          default=False,
      )
      return parser.parse_args()
  
file_path: ray/release/nightly_tests/dataset/gpu_batch_inference.py
imports: from typing import Dict
  import argparse
  import time
  import numpy
  import torch
  from torchvision.models import resnet50, ResNet50_Weights
  from benchmark import Benchmark, BenchmarkMetric
  import ray
  from ray.data import ActorPoolStrategy
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: parser