File size: 484 Bytes
c7ebaa1
 
 
 
 
 
bff2f94
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
"""Evaluation modules for BioRLHF."""

__all__ = [
    "evaluate_model",
    "compute_metrics",
]


def __getattr__(name):
    """Lazy imports for torch-dependent modules."""
    if name in ("evaluate_model", "compute_metrics"):
        from biorlhf.evaluation.evaluate import evaluate_model, compute_metrics
        return {"evaluate_model": evaluate_model, "compute_metrics": compute_metrics}[name]
    raise AttributeError(f"module 'biorlhf.evaluation' has no attribute {name!r}")