code: def get_dataset(
      dataset_name: str,
  ) -> Callable[..., dataset_utils.Dataset]:
      """Maps dataset name to a dataset_builder.
  
      API kept for compatibility of existing code with the DatasetRegistry.
  
      Args:
        dataset_name: Dataset name.
  
      Returns:
        A dataset builder.
      """
      return DatasetRegistry.get(dataset_name)
  
file_path: scenic/scenic/dataset_lib/datasets.py
imports: import functools
  import importlib
  from typing import Callable, List
  from absl import logging
  from scenic.dataset_lib import dataset_utils
label_desc: Use image node_id 2 for calling a global function with example usage: get_dataset(dataset_name) and returns: DatasetRegistry