code: def add_dataset(name: str, *args, **kwargs):
      """Decorator for shorthand dataset registdation."""
  
      def inner(
          builder_fn: Callable[..., dataset_utils.Dataset]
      ) -> Callable[..., dataset_utils.Dataset]:
          DatasetRegistry.add(
              name, functools.partial(builder_fn, *args, **kwargs)
          )
          return builder_fn
  
      return inner
  
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 1 for calling a global function with example usage: add_dataset(name) and returns: inner, builder_fn