code: def download_dataset(dir_path, nb_classes):
      if not os.path.exists(dir_path):
          os.makedirs(dir_path)
      assert os.path.isdir(
          dir_path
      ), f"{dir_path} exists and is not a directory"
  
      classes_path = os.path.join(
          os.path.dirname(__file__),
          "third_party/zaidalyafeai.github.io/class_names.txt",
      )
      with open(classes_path) as classes_file:
          classes = list(
              map(lambda c: c.strip(), classes_file.readlines())
          )[:nb_classes]
  
      url = "https://storage.googleapis.com/quickdraw_dataset/full/numpy_bitmap/"
      for cls in classes:
          cls_filename = f"{cls}.npy"
          cls_file_path = os.path.join(dir_path, cls_filename)
          if os.path.exists(cls_file_path):
              print(f"{cls_filename} already exists, skipping")
              continue
          with open(cls_file_path, "wb") as save_file:
              try:
                  response = requests.get(
                      url + cls_filename.replace("_", " ")
                  )
                  save_file.write(response.content)
                  print(f"Successfully fetched {cls_filename}")
              except:
                  print(f"Failed to fetch {cls_filename}")
      return classes
  
imports: from jax import numpy
  import numpy
  import os
  import requests
label_desc: Use image node_id 1 for calling a global function with example usage: download_dataset(dir_path, nb_classes) and returns: classes