code: def download_dataset(dataset_name):
      # code adapted from https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
      assert dataset_name in [
          "facades",
          "night2day",
          "edges2handbags",
          "edges2shoes",
          "maps",
      ]
      if os.path.exists("./data/" + dataset_name):
          _logger.info("Already downloaded dataset " + dataset_name)
      else:
          _logger.info("Downloading dataset " + dataset_name)
          if not os.path.exists("./data/"):
              pathlib.Path("./data/").mkdir(parents=True, exist_ok=True)
          pathlib.Path("./data/" + dataset_name).mkdir(
              parents=True, exist_ok=True
          )
          URL = "http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/{}.tar.gz".format(
              dataset_name
          )
          TAR_FILE = "./data/{}.tar.gz".format(dataset_name)
          TARGET_DIR = "./data/{}/".format(dataset_name)
          os.system("wget -N {} -O {}".format(URL, TAR_FILE))
          pathlib.Path(TARGET_DIR).mkdir(parents=True, exist_ok=True)
          os.system("tar -zxvf {} -C ./data/".format(TAR_FILE))
          os.system("rm " + TAR_FILE)
  
imports: import sys
  import os
  import pathlib
  import logging
  import time
  import argparse
  from collections import namedtuple
  import numpy
  import torch
  import torch.utils.data
  import nni
  from nni.utils import merge_parameter
  from pix2pixlib.data.aligned_dataset import AlignedDataset
  from pix2pixlib.models.pix2pix_model import Pix2PixModel
  from base_params import get_base_params
label_desc: Use image node_id 1 for calling a global function with example usage: download_dataset(dataset_name) without return types