code: def download_dataset(root: str):
      """
      Download the dSprites dataset archive and expand it in the folder provided as parameter
      """
      URL = "https://github.com/deepmind/dsprites-dataset/archive/master.zip"
      download_and_extract_archive(url=URL, download_root=root)
  
imports: import argparse
  import os
  from typing import Set
  import numpy
  from PIL import Image
  from torch.utils.data import DataLoader, Dataset
  from tqdm import tqdm
  from vissl.utils.download import download_and_extract_archive
label_desc: Use image node_id 2 for calling a global function with example usage: download_dataset(root) without return types