pff-pk-api / pff_pk /data /__init__.py
Dariusfar's picture
Support interactive synthetic cohorts, MLP resampling and configurable binned VPCs
a1a4d73 verified
Raw History Blame Contribute Delete
1.15 kB
"""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