code: def dataset(name: str, *args, **kwargs) -> "TorchDataSet":
      """Get a dataset by name, and pass into the custom params.
      Args:
          name (str): Name of a dataset.
          *args (any): Arguments of the dataset construct method.
          **kwargs (any): Keyword arguments of the dataset construct method.
      Returns:
          TorchDataSet: The corresponding `TorchDataSet`.
      Examples:
          >>> from towhee import dataset
          >>> type(dataset('fake', size=10))
          <class 'towhee.data.dataset.dataset.TorchDataSet'>
      """
      from torchvision import datasets
      from towhee.data.dataset.dataset import TorchDataSet
  
      dataset_construct_map = {
          "mnist": datasets.MNIST,
          "cifar10": datasets.cifar.CIFAR10,
          "fake": datasets.FakeData
          # 'imdb': IMDB  # ,()
      }
      torch_dataset = dataset_construct_map[name](*args, **kwargs)
      return TorchDataSet(torch_dataset)
  
file_path: towhee/towhee/__init__.py
imports: import sys
  from towhee.runtime import register, pipe, ops, accelerate, AutoConfig, AutoPipes
  from towhee.data_loader import DataLoader
  from towhee.serve.triton import triton_client
  from towhee.utils.lazy_import import LazyImport
  from towhee import types
label_desc: Use image node_id 4 for calling a global function with example usage: dataset(name) and returns: TorchDataSet