Spaces:
Running
Running
File size: 1,001 Bytes
877add7 | 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 30 31 32 33 34 35 36 37 | from app.training.reward_functions import (
abstention_quality_score,
anti_cheat_score,
candidate_alignment_score,
disease_stability_score,
dosing_quality_score,
efficiency_score,
explanation_grounding_score,
format_compliance_score,
legality_score,
process_fidelity_score,
safety_delta_score,
uncertainty_calibration_score,
)
def _in_range(v: float) -> bool:
return 0.001 <= v <= 0.999
def test_all_reward_functions_range() -> None:
values = [
format_compliance_score(True),
candidate_alignment_score(True),
legality_score(True),
safety_delta_score(0.2),
disease_stability_score(0.8),
dosing_quality_score(0.7),
abstention_quality_score(True),
efficiency_score(0.4),
process_fidelity_score(0.9),
explanation_grounding_score(0.75),
anti_cheat_score(False),
uncertainty_calibration_score(0.65),
]
assert all(_in_range(v) for v in values)
|