code: def register_dataset(
      cls, dataset_name: str
  ) -> _typing.Callable[[_typing.Type[Dataset]], _typing.Type[Dataset]]:
      def register_dataset_cls(dataset: _typing.Type[Dataset]):
          # print(dataset_name)
          if (not callable(dataset)) and (
              not issubclass(dataset, Dataset)
          ):
              raise TypeError
          else:
              cls[dataset_name] = dataset
              return dataset
  
      return register_dataset_cls
  
file_path: AutoGL/autogl/datasets/_dataset_registry.py
imports: import os
  import typing
  from autogl.utils import universal_registry
  from autogl.data import Dataset
label_desc: Use image node_id 1 for calling the DatasetUniversalRegistry obj's underlying member method code with example usage: obj.register_dataset(cls, dataset_name) and returns: register_dataset_cls, dataset