code: def register_dataset(name: str, config: DatasetConfig):
      _REGISTRY[name] = config
  
file_path: weak-to-strong/weak_to_strong/datasets.py
imports: import functools
  from dataclasses import dataclass
  from random import Random
  from typing import Any, Callable, Optional
  from datasets import Dataset
  from datasets import load_dataset
label_desc: Use image node_id 1 for calling a global function with example usage: register_dataset(name, config) without return types