haiku.rag/evaluations/evaluations/datasets/__init__.py

8 lines
216 B
Python

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