code: def load_dataset(
      name: str,
  ) -> DATASET_TYPE:
      """
      Loads or downloads the dataset corresponding to `name`. Options are: `mnist`, `cifar10`, `stl10`, `iris`, `nursery`
      and `diabetes`.
  
      :param name: Name of the dataset.
      :return: The dataset separated in training and test sets as `(x_train, y_train), (x_test, y_test), min, max`.
      :raises NotImplementedError: If the dataset is unknown.
      """
      if "mnist" in name:
          return load_mnist()
      if "cifar10" in name:
          return load_cifar10()
      if "stl10" in name:
          return load_stl()
      if "iris" in name:
          return load_iris()
      if "nursery" in name:
          return load_nursery()
      if "diabetes" in name:
          return load_diabetes()
  
      raise NotImplementedError(
          f"There is no loader for dataset '{name}'."
      )
  
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  import logging
  import math
  import os
  import shutil
  import sys
  import tarfile
  import warnings
  import zipfile
  from functools import wraps
  from inspect import signature
  from typing import TYPE_CHECKING, Callable, List, Dict, Optional, Tuple, Union
  import numpy
  import six
  from scipy.special import gammainc
  from tqdm.auto import tqdm
  from art import config
label_desc: Use image node_id 34 for calling a global function with example usage: load_dataset(name) and returns: load_mnist, load_cifar10, load_stl, load_iris, load_nursery, load_diabetes