"""Corpus access and contracts; historical convenience exports load lazily."""

from importlib import import_module

_EXPORTS = {
    "StreamingStudyTaskDataset": "pff_pk.data.corpus",
    "StudyTaskDataset": "pff_pk.data.corpus",
    "discover_shards": "pff_pk.data.corpus",
    "load_study_blobs": "pff_pk.data.corpus",
    "load_split_shards": "pff_pk.data.splits",
    "DoseEvent": "pff_pk.data.studies",
    "Individual": "pff_pk.data.studies",
    "Study": "pff_pk.data.studies",
    "StudyFamily": "pff_pk.data.studies",
    "parse_protocol_family": "pff_pk.data.studies",
    "parse_record": "pff_pk.data.studies",
    "parse_study": "pff_pk.data.studies",
    "FlowBatch": "pff_pk.tasks.batch",
    "FlowTask": "pff_pk.tasks.batch",
    "collate_tasks": "pff_pk.tasks.batching",
    "inverse_concentration": "pff_pk.tasks.batching",
    "sample_task": "pff_pk.tasks.generation",
}
__all__ = list(_EXPORTS)


def __getattr__(name):
    if name not in _EXPORTS:
        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
    value = getattr(import_module(_EXPORTS[name]), name)
    globals()[name] = value
    return value