haiku.rag/evaluations/evaluations/datasets/__init__.py
2026-06-01 10:40:51 +03:00

15 lines
310 B
Python

from evaluations.config import DatasetSpec
from .open_rag_bench import ORB_MULTIMODAL_SPEC, ORB_TEXT_SPEC
from .wix import WIX_SPEC
DATASETS: dict[str, DatasetSpec] = {
spec.key: spec
for spec in (
WIX_SPEC,
ORB_TEXT_SPEC,
ORB_MULTIMODAL_SPEC,
)
}
__all__ = ["DATASETS"]