"""按配置构建 Dataset。"""

from __future__ import annotations

from collections.abc import Mapping
from typing import Any

from torch.utils.data import Dataset

from datasets.registry import DATASETS


def build_dataset(cfg: Mapping[str, Any], **kwargs: Any) -> Dataset:
    return DATASETS.build(dict(cfg), **kwargs)