code: def parse_args():
      parser = argparse.ArgumentParser(
          description="Preprocess criteo dataset"
      )
      parser.add_argument(
          "--base_path", "-b", dest="base_path", help="Base path"
      )
      parser.add_argument(
          "--shuffle_train",
          "-s",
          dest="shuffle_train",
          default=False,
          action="store_true",
          help="Base path",
      )
      args = parser.parse_args()
  
      return args
  
file_path: torchrec/torchrec/datasets/scripts/nvt/process_criteo_parquet.py
imports: import argparse
  import os
  import shutil
  import time
  import numpy
  import nvtabular
  from merlin.io import Shuffle
  from utils.criteo_constant import DAYS, DEFAULT_CAT_NAMES, DEFAULT_COLUMN_NAMES, DEFAULT_INT_NAMES, DEFAULT_LABEL_NAME, NUM_EMBEDDINGS_PER_FEATURE_DICT
  from utils.dask import setup_dask
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args