registry

def register_dataset(name: str):
    """Decorator to register a prompt-dataset class.
Usage::

    @register_dataset("vbench")
    class VBenchPromptDataset(BasePromptDataset):
        ...
"""

def wrapper(cls):
    cls.name = name
    _REGISTRY[name] = cls
    return cls

return wrapper</code></pre></details>
fastvideo.eval.datasets.registry.get_dataset
get_dataset(name: str, **kwargs: Any) -> BasePromptDataset
Instantiate a registered dataset by name.
Source
def get_dataset(name: str, **kwargs: Any) -> BasePromptDataset:
    """Instantiate a registered dataset by name."""
    cls = _REGISTRY.get(name)
    if cls is None:
        available = ", ".join(sorted(_REGISTRY.keys()))
        raise KeyError(f"Unknown dataset '{name}'. Available: {available}")
    return cls(**kwargs)
fastvideo.eval.datasets.registry.list_datasets
list_datasets() -> list[str]
Return sorted list of all registered dataset names.
Source
def list_datasets() -> list[str]:
    """Return sorted list of all registered dataset names."""
    return sorted(_REGISTRY.keys())