Spaces:
Sleeping
Sleeping
File size: 711 Bytes
aeba3f1 | 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 | """Reward functions exposed for TRL/GRPO logging."""
def _values(name: str, completions, kwargs):
return [float(x) for x in kwargs.get(name, [0.0] * len(completions))]
def reward_total(completions, **kwargs):
return _values("reward_total", completions, kwargs)
def reward_security(completions, **kwargs):
return _values("reward_security", completions, kwargs)
def reward_regression(completions, **kwargs):
return _values("reward_regression", completions, kwargs)
def reward_patch_quality(completions, **kwargs):
return _values("reward_patch_quality", completions, kwargs)
def reward_anti_cheat(completions, **kwargs):
return _values("reward_anti_cheat", completions, kwargs)
|