File size: 820 Bytes
c7ebaa1 bff2f94 c7ebaa1 bff2f94 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | """Data processing and dataset creation modules for BioRLHF."""
# ground_truth has no heavy dependencies, safe to import eagerly
from biorlhf.data.ground_truth import (
STRESSOR_EFFECTS,
KMP_EFFECTS,
INTERACTIONS,
TISSUE_TYPES,
OXPHOS_PATTERNS,
)
__all__ = [
"create_sft_dataset",
"load_dataset",
"STRESSOR_EFFECTS",
"KMP_EFFECTS",
"INTERACTIONS",
"TISSUE_TYPES",
"OXPHOS_PATTERNS",
]
def __getattr__(name):
"""Lazy imports for modules with heavy dependencies."""
if name in ("create_sft_dataset", "load_dataset"):
from biorlhf.data.dataset import create_sft_dataset, load_dataset
return {"create_sft_dataset": create_sft_dataset, "load_dataset": load_dataset}[name]
raise AttributeError(f"module 'biorlhf.data' has no attribute {name!r}")
|