code: def register_dataset(name: str) -> Callable[[D], D]:
      def wrapper(dataset_cls: D) -> D:
          BUILTIN_DATASETS[name] = dataset_cls
          return dataset_cls
  
      return wrapper
  
file_path: vision/torchvision/prototype/datasets/_api.py
imports: import pathlib
  from typing import Any, Callable, Dict, List, Optional, Type, TypeVar, Union
  from torchvision.prototype.datasets import home
  from torchvision.prototype.datasets.utils import Dataset
  from torchvision.prototype.utils._internal import add_suggestion
label_desc: Use image node_id 2 for calling a global function with example usage: register_dataset(name) and returns: wrapper, dataset_cls