tau2 scorer: read the composite 'reward' field
tau2's current reward_info carries 'reward' (composite) + db_check / action_checks; the scorer read the old environment_reward / communication_reward split that no longer exists -- simulations scored 1.0 came out 0.0. Fallback to the old split kept for older engines. Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
parent
15b1574857
commit
f844775f1e
@ -262,16 +262,23 @@ def swe_bench_verified():
|
|||||||
|
|
||||||
|
|
||||||
def _tau2_reward(pred, target, sample, ctx):
|
def _tau2_reward(pred, target, sample, ctx):
|
||||||
"""Score from the official engine's reward_info (env_state)."""
|
"""Score from the official engine's reward_info (env_state).
|
||||||
|
|
||||||
|
Current tau2 reward_info carries the COMPOSITE 'reward' plus detail
|
||||||
|
fields (db_check / action_checks / communicate_checks); the old
|
||||||
|
environment_reward/communication_reward split no longer exists."""
|
||||||
env_state = ctx.params.get('env_state') or {}
|
env_state = ctx.params.get('env_state') or {}
|
||||||
rewards = env_state.get('tau2_rewards') or {}
|
rewards = env_state.get('tau2_rewards') or {}
|
||||||
env_r = rewards.get('environment_reward')
|
r = rewards.get('reward')
|
||||||
comm_r = rewards.get('communication_reward')
|
if not isinstance(r, (int, float)):
|
||||||
vals = [r for r in (env_r, comm_r) if isinstance(r, (int, float))]
|
vals = [v for v in (rewards.get('environment_reward'),
|
||||||
score = float(sum(vals) / len(vals)) if vals else 0.0
|
rewards.get('communication_reward'))
|
||||||
return ({'acc': score}, {'acc': {'mode': 'official_tau2',
|
if isinstance(v, (int, float))]
|
||||||
'env_reward': env_r,
|
r = sum(vals) / len(vals) if vals else 0.0
|
||||||
'comm_reward': comm_r}})
|
return ({'acc': float(r)}, {'acc': {'mode': 'official_tau2',
|
||||||
|
'reward': r,
|
||||||
|
'db_check': rewards.get('db_check'),
|
||||||
|
'note': str((rewards.get('info') or {}).get('note', ''))[:120]}})
|
||||||
|
|
||||||
|
|
||||||
@register_eval('tau2_bench')
|
@register_eval('tau2_bench')
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user