code: def main():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--no-generate", default=False, action="store_true"
      )
      parser.add_argument("--sas-token", default=None, type=str)
      parser.add_argument("--data-dir", default="./data", type=str)
      parser.add_argument(
          "--imagenet-dir", default="/mnt/data/imagenet", type=str
      )
  
      args = parser.parse_args()
      data_dir = Path(args.data_dir)
      if args.no_generate:
          print("Generation is skipped.")
      else:
          prepare_cifar10(data_dir)
          prepare_mnist(data_dir)
          prepare_imagenet_subset(data_dir, Path(args.imagenet_dir))
          zip_datasets(data_dir)
      upload_datasets(args.sas_token)
  
file_path: nni/test/vso_tools/pack_testdata.py
imports: import argparse
  import glob
  import os
  import random
  import shutil
  import subprocess
  import warnings
  from collections import defaultdict
  from pathlib import Path
  from torchvision.datasets import CIFAR10, MNIST, ImageNet
label_desc: Use image node_id 6 for calling a global function with example usage: main() without return types