| """FATHOM reward package — REW-01. | |
| Exposes: | |
| - format_gate(completion, **_) -> float | |
| - correctness(completion, gold_answer, **_) -> float | |
| - token_budget(completion, prompt_token_count, alpha, variant, **_) -> float | |
| - recursion_efficiency(llm_call_count, max_calls, **_) -> float | |
| - compose_reward_fn(prompts, completions, **kwargs) -> list[float] [TRL contract] | |
| - compose_reward_single(completion, gold_answer, prompt_token_count, llm_call_count, cfg_reward) -> float | |
| - make_reward_fn(cfg_reward) -> callable | |
| """ | |
| from .format_gate import format_gate | |
| from .correctness import correctness | |
| from .token_budget import token_budget | |
| from .recursion_efficiency import recursion_efficiency | |
| from .compose import compose_reward_fn, compose_reward_single, make_reward_fn | |
| __all__ = [ | |
| "format_gate", | |
| "correctness", | |
| "token_budget", | |
| "recursion_efficiency", | |
| "compose_reward_fn", | |
| "compose_reward_single", | |
| "make_reward_fn", | |
| ] | |