code: def parse_args(argv: List[str]) -> argparse.Namespace:
      parser = argparse.ArgumentParser(
          description="Criteo tsv -> npy preprocessing script."
      )
      parser.add_argument(
          "--input_dir",
          type=str,
          required=True,
          help="Input directory containing Criteo tsv files."
          "For criteo_1tb, files in the directory should be named day_{0-23}."
          "For criteo_kaggle, files in the directory should be train.txt & test.txt.",
      )
      parser.add_argument(
          "--output_dir",
          type=str,
          required=True,
          help="Output directory to store npy files.",
      )
      parser.add_argument(
          "--dataset_name",
          type=str,
          choices=["criteo_1tb", "criteo_kaggle"],
          default="criteo_1tb",
          help="dataset for experiment, current support criteo_1tb, criteo_kaggle",
      )
      return parser.parse_args(argv)
  
imports: import argparse
  import os
  import sys
  from typing import List
  from torchrec.datasets.criteo import BinaryCriteoUtils
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args(argv) and returns: parser